2021年7月21日
torch.argmax中dim详解
摘要： torch.argmax()函数 argmax函数：torch.argmax(input, dim=None, keepdim=False)返回指定维度最大值的序号，dim给定的定义是：the demention to reduce.也就是把dim这个维度的，变成这个维度的最大值的index。 例如 阅读全文
posted @ 2021-07-21 18:17 尘埃02 阅读(2871) 评论(0) 推荐(1) 编辑