📌  相关文章
📜  pytorch 张量 argmax - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:24.756000             🧑  作者: Mango

代码示例1
values, indices = tensor.max(0)
values, indices = torch.max(tensor, 0)