luwanglin

摘要： pytorch常用函数总结（持续更新) torch.max(input,dim) 求取指定维度上的最大值，,返回输入张量给定维度上每行的最大值，并同时返回每个最大值的位置索引。比如： demo.shape Out[7]: torch.Size([10, 3, 10, 10]) torch.max(d 阅读全文