pytorch max和clamp

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,可用于一个数和数组比较,用最大或者最小值替换数组对应位置的值

猜你喜欢

转载自www.cnblogs.com/lzping/p/12362474.html