質問編集履歴
1
自分で試したこと追加
test
CHANGED
File without changes
|
test
CHANGED
@@ -2,6 +2,8 @@
|
|
2
2
|
Pytorchで3次元のtensorがあり、tensor中のdim=1の中で値が一番大きい場所を1に、一番大きい値以外を0にしたいのですがどうすればいいでしょうか?
|
3
3
|
|
4
4
|
ご教授よろしくお願いします。
|
5
|
+
|
6
|
+
torch.argmaxやtorch.max、スライス処理など色々考えてみたんですがわからなかったです、、、
|
5
7
|
|
6
8
|
### 該当のソースコード
|
7
9
|
|