torch.meshgrid和np.meshgrid的区别

numpy中meshgrid:
把数组a当作一行,再根据数组b的长度扩充行。
把数组b当作一列,再根据数组a的长度扩充列。

torch中meshgrid:
把数组a当作一列,再根据数组b的长度扩充列。
把数组b当作一行,再根据数组a的长度扩充行。

torch.meshgrid和np.meshgrid的区别_第1张图片

你可能感兴趣的:(python,深度学习)