回答編集履歴
1
実際の出力を記載
test
CHANGED
|
@@ -6,4 +6,8 @@
|
|
|
6
6
|
|
|
7
7
|
|
|
8
8
|
|
|
9
|
-
xとattnのshapeを画面出力させて確認しました．
|
|
9
|
+
xとattnのshapeを画面出力させて確認しました．
|
|
10
|
+
|
|
11
|
+
x.shape: torch.Size([32, 512, 7, 7])
|
|
12
|
+
|
|
13
|
+
attn.shape: torch.Size([32, 1, 7, 7])
|