返回最大值的indexpytorch方式目录返回最大值的indexpytorch 找最大值返回最大值的index
import torch
a=torch.tensor([[.1,.2,.3],