torch.argmax()
torch.argmax(input, dim=None, keepdim=False)转载至 : https://blog.csdn.net/qq_46092061/article/details/120612102torch.argmax(input, dim=None, keepdim=False)返回指定维度最大值的序号index函数中dim表示该维度会消失。dim=0,表示我们不要行了,求出每一列的最大值的下标;dim=1,表示我们不要列了,保留每一行的最大值的下标。二维例子
转载
2021-11-17 15:42:51 ·
491 阅读 ·
0 评论