Ich arbeite an der Durchführung von CNN Inferenz für MNIST auf iOS. Es gibt ein gutes Codebeispiel von Apple, um zu starten. https://developer.apple.com/library/content/samplecode/MPSCNNHelloWorld/Introduction/Intro.html#//apple_ref/doc/uid/TP40017482-Intro-DontLinkElementID_2Wie mache ich einen Flatten-Filter mit MPS (MetalPerformanceShaders)?
jedoch Als ich versuchte, viel komplexes CNN-Modell (z https://github.com/fchollet/keras/blob/master/examples/mnist_cnn.py) zu implementieren, indem MPS verwendet, fand ich, dass es keine Klasse für „Flatten“ Filter war.
Ich schaute durch MPS-Framework, um die Funktion zum Umformen oder Ändern einer Dimension zu finden, aber ich konnte nicht die richtige finden. (ZB MPSImageConversion scheint nur für eine Farbe Umwandlung, aber nicht für eine Dimension.
Wenn jemand die Filter für Flatten weiß oder wie 1D Bild mehrdimens Bild konvertieren, lass es mich wissen.
Ich fand, dass Flatten-Layer nicht erforderlich ist, da MPS mehrdimensionale Daten als Eingabe verarbeiten kann. – kangaroo