ios - 如何使用 MPS (MetalPerformanceShaders) 制作 Flatten 滤镜？

Question

我正在为 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 的过滤器或如何将多维图像转换为一维图像，请告诉我。

Answer 1

不需要 Flatten 层，因为 MPS 可以处理多维数据作为输入。