質問編集履歴

1

誤字

2023/03/21 09:31

投稿

tanhxyx
tanhxyx

スコア4

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