pytorch输出numarry最大值下标

在PyTorch中，如果你有一个张量（tensor）并且想要找出其中数值的最大值及其对应的索引，你可以使用`torch.argmax()`函数。这个函数返回的是第一个元素最大值的索引，如果你想获取所有最大值的索引，可以配合`torch.stack()`或者`torch.nonzero()`一起使用。

例如：

import torch

# 假设我们有一个一维张量 tensor
input_tensor = torch.tensor([3, 5, 1, 9, 2])

# 获取最大值的索引
max_index = torch.argmax(input_tensor)

print("最大值索引:", max_index.item())  # 输出：4，因为9是最大值，其索引是4

# 如果你想找到所有的最大值索引，可以这样做：
all_max_indices = torch.nonzero(input_tensor == input_tensor.max()).squeeze(1)
print("所有最大值索引:", all_max_indices.tolist())