torch.max()和np.max()的一点区别,补充np.maximum()

除了网上所说torch.max()除返回最大值还返回索引,而np.max只返回最大值以外,还有一个很大的区别在于,torch.max可以直接比较两个张量,也就是torch.max(a,b)是可以的,但是np.max只能有一个输入,当要比较两个数组的术后,要采用np.max([a,b])的形式

但np.maximum天然用于比较两个数组

另外一个特别重要的点在于:torch.max()和np.maximum都支持广播机制,但np.max不支持广播机制。

实验如下:

torch.max()和np.max()的一点区别,补充np.maximum()_第1张图片

np.max()只能输入一个值

torch.max()和np.max()的一点区别,补充np.maximum()_第2张图片

np.maximum()天然比较两个数组

torch.max()和np.max()的一点区别,补充np.maximum()_第3张图片

np.maximum和torch.max支持数据的广播机制,但np.max()不支持数据广播

torch.max()和np.max()的一点区别,补充np.maximum()_第4张图片

这里一个shape(3,3),一个shape(2,3)我们在(2,3)这里扩展一个维度变为(2,1,3)所以通过广播以后结果是(2,3,3)的shape。

 torch.max()和np.max()的一点区别,补充np.maximum()_第5张图片

 

 

你可能感兴趣的:(CV笔记,pytorch,深度学习,机器学习)