2017-01-15 1 views
1

Gibt es eine Ebene (oder eine einfache Kombination von Ebenen), die zur Matrixmultiplikation zwischen zwei Feature-Maps verwendet werden kann? Angenommen, es gibt einen Blob mit der Größe MxN und einen anderen mit der Größe NxP. Gibt es eine Ebene, die diese beiden Blobs wie eine Matrix multiplizieren kann?Matrixmultiplikation zwischen zwei Feature-Maps in Caffe

Antwort

1

Es gibt eine geplante Änderung von caffe, um Manipulationen während der Abfrage zu ermöglichen, dh Parameterblobs als reguläre Blobs zu behandeln. Weitere Informationen finden Sie unter this answer.

Bis diese Änderung abgeschlossen ist, können Sie

  1. "Tile" ing die Blobs in M x N x P Blobs betrachten.
  2. Verwendung der "Eltwise" Schicht für die elementweise Multiplikation.
  3. Verwendung von "Pooling" mit globalen Pool auf der N Achse.
+1

Ich dachte daran, aber es dauert viel mehr Speicher (linear mehr). Ich benutzte caffe_gpu_gemm, weil ich keinen Rückwärtsdurchlauf brauchte. – Bharat