文章比较了在PyTorch中使用torch.max和torch.argmax两个函数获取最大值索引时的执行速度。实测结果显示torch.max比torch.argmax快约一秒，特别是在处理大型数据时，这种差异可能具有实际意义。
解决方法如下：
_, y_hat = torch.max(y_hat, 1)
# or
y_hat = torch.argmax(y_hat.to('cpu'), axis=1).to('mps')
实测，第一个快一秒
Mac m1 torch.argmax 只返回 -9223372036854775
于 2023-04-05 21:01:22 首次发布
解决方法如下：
_, y_hat = torch.max(y_hat, 1)
# or
y_hat = torch.argmax(y_hat.to('cpu'), axis=1).to('mps')
实测，第一个快一秒
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
