我正在为 iOS 上的 MNIST 运行 CNN 推理。 Apple 提供了一个很好的代码示例作为开始。 https://developer.apple.com/library/content/samplecode/MPSCNNHelloWorld/Introduction/Intro.html#//apple_ref/doc/uid/TP40017482-Intro-DontLinkElementID_2
但是,当我尝试使用 MPS 实现更复杂的 CNN 模型(例如 https://github.com/fchollet/keras/blob/master/examples/mnist_cnn.py )时,我发现没有“Flatten”过滤器的类。
我查看了 MPS 框架以找到 reshape 或更改维度的函数,但找不到合适的函数。 (例如 MPSImageConversion
似乎只用于转换颜色,而不是尺寸。
如果有人知道 Flatten 的过滤器或如何将多维图像转换为一维图像,请告诉我。
最佳答案
不需要 Flatten 层,因为 MPS 可以处理多维数据作为输入。
关于ios - 如何使用 MPS (MetalPerformanceShaders) 制作 Flatten 滤镜?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40814481/