torch类似argmax的函数是哪个?

torch类似argmax的函数是哪个?
已邀请:

Also

赞同来自:

torch.max会返回两个变量,第一个是最大的数值,第二个就是下标

要回复问题请先登录注册