回答編集履歴
2
test
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
numpyもそうですが、元配列に対するbooleanな結果を出す式を「マスク配列」と呼びます。
|
|
2
|
-
例えば[True, True, False]みたいな
|
|
2
|
+
例えば[True, True, False]みたいな配列がマスク配列です。
|
|
3
3
|
|
|
4
4
|
**元の配列[マスク配列] = 一致したときに置き換えたい値**
|
|
5
5
|
という構造でやりたいことができます。
|
1
test
CHANGED
|
@@ -1,4 +1,3 @@
|
|
|
1
|
-
走らせてませんがたぶん動きます。
|
|
2
1
|
numpyもそうですが、元配列に対するbooleanな結果を出す式を「マスク配列」と呼びます。
|
|
3
2
|
例えば[True, True, False]みたいな結果が得られます。
|
|
4
3
|
|
|
@@ -7,6 +6,7 @@
|
|
|
7
6
|
|
|
8
7
|
参考： [この記事の「リスト1.3 Tensorのインデクシング操作例」](https://codezine.jp/article/detail/11052)を読んでください。
|
|
9
8
|
|
|
9
|
+
以下、走らせてませんがたぶん動きます。
|
|
10
10
|
```Python3
|
|
11
11
|
inp1 = torch.rand((1, 5), dtype=torch.float32)
|
|
12
12
|
idx = torch.argmax(inp1)
|