回答編集履歴
1
実際の出力を記載
answer
CHANGED
|
@@ -2,4 +2,6 @@
|
|
|
2
2
|
|
|
3
3
|
画像では入力画像と同じサイズになっていますが，それは可視化した場合．実際には`バッチサイズ×チャネル×タテ×ヨコ = 32×1×7×7`となっていました．
|
|
4
4
|
|
|
5
|
-
xとattnのshapeを画面出力させて確認しました．
|
|
5
|
+
xとattnのshapeを画面出力させて確認しました．
|
|
6
|
+
x.shape: torch.Size([32, 512, 7, 7])
|
|
7
|
+
attn.shape: torch.Size([32, 1, 7, 7])
|