Erreichen tf in Metall Performance Shader

Question

umsetzen, was der beste Weg ist, um die folgenden in Metall

inputs_transposed = tf.transpose(inputs,(2,0,1))

Dank Hollemans zu erreichen, ich weiß, das ist ein Rechenkern sein muss, aber ich bin nicht sicher, wie man Schreib es. Hat jemand versucht, in Metal zu transponieren, wenn der Eingang mehr als 4 Feature-Kanäle hat.

Answer 1

Sie möchten Ihre Daten während der Übertragung transponieren? Dann müssen Sie tatsächlich einen Rechenkernel schreiben.

Es wäre in Metall Shading Language in etwa so aussehen:

half4 inColor = inTexture.read(gid); 
half4 outColor = half4(inColor[2], inColor[0], inColor[1], 0.0h); 
outTexture.write(outColor, gid); 

Wo gid Punkte auf Pixel, dass diese besondere Kernel arbeitet.

Um dies für mehr als 4 Kanäle zu tun, ist es ziemlich das Gleiche, aber jetzt müssen Sie aus verschiedenen Textur Slices im Kernel lesen/schreiben.

Das obige Beispiel zeigt Ihre (2, 0, 1) transponieren, aber hardcoded in den Kernel. Es wird etwas komplizierter, wenn Sie auf beliebige Weise transponieren möchten. In diesem Fall müssen Sie die neue Kanalreihenfolge unter Verwendung einer MTLBuffer in den Kernel übergeben, und Sie müssen herausfinden, welcher Kanalindex welchem ​​Texturabschnitt entspricht und so weiter.

Ich weiß, das ist immer noch eine vage Antwort. Ich werde sehen, ob ich dafür einen Beispielkernel erstellen und zu meiner Open-Source-Bibliothek Forge hinzufügen kann. https://github.com/hollance/Forge

Edit: Ich habe einen einfachen transponieren Kernel zu schmieden: https://github.com/hollance/Forge/blob/master/Forge/Forge/TransposeChannels.swift

Sie verwenden es wie folgt aus:

let channels = 8 
let perm: [UInt16] = [2, 4, 0, 5, 1, 7, 6, 3] 
transposeKernel = TransposeChannelsKernel(device: device, 
          featureChannels: channels, permute: perm) 

Und dann codieren Sie es wie jedes andere MPSCNN kernel:

transposeKernel.encode(commandBuffer: commandBuffer, 
         sourceImage: img0, destinationImage: img1) 

Da es ein Teil von Forge ist, verwendet es eine Reihe von Helfer-Code aus dieser Bibliothek. Wenn Sie es zu Ihrem eigenen Projekt hinzufügen und Forge nicht verwenden möchten, müssen Sie auch diesen Hilfscode kopieren.

Edit 2: Von Ihrem Kommentar unten scheint es, wie Sie nicht nur Kanäle wollen um tauschen, sondern ganze Achsen, so dass dieses TransposeChannels Objekt wird nicht auf Sie zu sehr nützlich. Hey, zumindest habe ich meiner Bibliothek Code hinzugefügt. :-D