2017-01-15 9 views
1

2つのフィーチャマップ間で行列乗算を実行するために使用できるレイヤ(またはレイヤの単純な組み合わせ)はありますか?たとえば、ブロブのサイズがMxNで、サイズがNxPのブロブがあるとします。これらのブロブを行列のように乗算するレイヤーはありますか?Caffeの2つのフィーチャマップ間の行列乗算

答えて

1

あなたが聞いているように操作を可能にするために、パラメータblobを規則的なblobとして扱う予定の変更が予定されています。詳細については、this answerを参照してください。

この変更が完了するまで、あなたはM X N X P塊にブロブをINGの

  1. "Tile"を考えるかもしれません。
  2. 要素ごとの乗算のために"Eltwise"層を使用する。
  3. "Pooling"をグローバルプールにN軸で使用しています。
+1

私はこれを考えましたが、より多くのメモリ(直線的に多く)が必要です。バックパスは必要ないので、私はcaffe_gpu_gemmを使用して終了しました。 – Bharat