2017-01-15 2 views
1

두 기능 맵간에 행렬 곱셈을 수행하는 데 사용할 수있는 레이어 (또는 레이어의 간단한 조합)가 있습니까? 예를 들어, BLOB의 크기가 MxN이고 다른 크기가 NxP 인 경우 BLOB에 행렬이 곱해질 수있는 레이어가 있습니까?Caffe의 두 기능 맵 간의 매트릭스 곱

답변

1

매개 변수 blob을 규칙적인 blob로 처리하도록 요청할 때 조작을 허용하기 위해 계획된 caffe 변경이 있습니다. 자세한 내용은 this answer을 참조하십시오. 이 변경이 완료 될 때까지

, 당신은 M X N X P 모양으로 모양을 보내고

  1. "Tile"을 고려할 수 있습니다.
  2. 요소 - 와이즈 곱셈에 대해 "Eltwise" 층을 사용함.
  3. "Pooling"을 축 풀 N 축에 사용합니다.
+1

나는 이것에 대해 생각했지만 훨씬 더 많은 메모리 (선형 적으로)를 필요로한다. 나는 후방 통과가 필요 없기 때문에 caffe_gpu_gemm을 사용하여 끝냈다. – Bharat