質問編集履歴
1
誤字
test
CHANGED
File without changes
|
test
CHANGED
@@ -15,13 +15,13 @@
|
|
15
15
|
普通に実装すると以下のようになるのですがfor文を使わず関数で綺麗に書けないでしょうか
|
16
16
|
|
17
17
|
```python
|
18
|
-
out = torch.zeros((index.size(0), data.size(1))
|
18
|
+
out = torch.zeros((index.size(0)-1, data.size(1))
|
19
19
|
for i in range(index.size(0)-1):
|
20
20
|
out[i] = torch.mean(data[index[i]:index[i+1], dim=0)
|
21
21
|
```
|
22
22
|
|
23
23
|
```python
|
24
|
-
out = torch.zeros((index.size(0), data.size(1))
|
24
|
+
out = torch.zeros((index.size(0)-1, data.size(1))
|
25
25
|
target = torch.zeros_like(index)
|
26
26
|
for i in range(index.size(0)-1):
|
27
27
|
target[index[i]:index[i+1]]=i
|