2021年3月28日
argmax()、max()介绍
摘要： 在开始之前先明确“行轴”和“列轴”的方向如上图。图中“行轴”索引范围为0-2，轴长为3，“列轴”索引范围0-3，轴长为4。 对张量的操作 import torch t = torch.tensor([[3,4,1,0],[12,1,2,3],[52,34,88,11]]) t # 输出为： tens 阅读全文
posted @ 2021-03-28 19:29 rain-fallz 阅读(434) 评论(0) 推荐(0)