pytorch max和clamp

标签：amp   ORC   pytorch   tor   维数   数组   大小   个数   max   

torch.max()

torch.max(a)：数组a的最大值
torch.max(a, dim=1)：多维数组沿维度1方向上的最大值，若a为二维数组，则为每行的最大值(此时是对每行的每列值比较取最大，即沿列的方向最大值)
torch.max(a,b)：对同样大小的两个数组比较，取对应位置上的最大值

torch.clamp()

torch.clamp(a, min, max)：对数组a的每个元素，超过max的取max,小于min的取min,可用于一个数和数组比较，用最大或者最小值替换数组对应位置的值

pytorch max和clamp

标签：amp   ORC   pytorch   tor   维数   数组   大小   个数   max   

原文地址：https://www.cnblogs.com/lzping/p/12362474.html