torch.argmax dim=1 和axis
时间: 2024-03-30 14:38:20 浏览: 145
浅谈pytorch torch.backends.cudnn设置作用
5星 · 资源好评率100%
torch.argmax(dim=1)和axis的作用是一样的,都是在指定维度上取最大值的索引。在PyTorch中,dim是指定维度的参数,而在NumPy和Pandas中,axis是指定维度的参数。两者的用法类似,只是名称不同。例如,对于一个2维的张量,dim=0表示在行上进行操作,dim=1表示在列上进行操作。
阅读全文