Row with Highest Mean
0
Compute the mean for each row of a 2D tensor and return the index of the row with the highest mean value. For example, if your tensor is `[[1, 2, 3], [4, 5, 6], [7, 8, 9]]`, the means for each row are `[2, 5, 8]`. Thus, the expected output should be `2`, as the third row (0-indexed) has the highest mean.
To solve this exercise, first compute the mean of each row using `torch.mean(tensor, dim=1)`. This will give you a 1D tensor with the means of each row. Then, use `torch.argmax()` to get the index of the highest mean value.Examples:
1.0
2
3
4
5
6
7
8
9
↓
2
10.0
20
30
40
50
60
↓
2
Loading...