Je travaille sur l'inférence CNN pour MNIST sur iOS. Il y a un bon exemple de code d'Apple pour commencer. https://developer.apple.com/library/content/samplecode/MPSCNNHelloWorld/Introduction/Intro.html#//apple_ref/doc/uid/TP40017482-Intro-DontLinkElementID_2Comment faire un filtre Flatten en utilisant MPS (MetalPerformanceShaders)?
Cependant, Quand j'ai essayé de mettre en œuvre le modèle CNN beaucoup plus complexe (par exemple https://github.com/fchollet/keras/blob/master/examples/mnist_cnn.py) en utilisant MPS, j'ai trouvé qu'il n'y avait pas de classe pour le filtre « Aplatir ». J'ai regardé à travers le cadre MPS pour trouver la fonction pour remodeler ou modifier une dimension, mais je ne pouvais pas trouver la bonne. (Par exemple MPSImageConversion semble que pour la conversion d'une couleur, mais pas pour une dimension.
Si quelqu'un connaît le filtre pour Aplatir ou comment convertir l'image multi-dimension à l'image 1D, s'il vous plaît laissez-moi savoir.
J'ai trouvé que la couche Flatten n'est pas nécessaire car MPS peut gérer des données multidimensionnelles en entrée. – kangaroo