numpy中meshgrid:
把数组a当作一行,再根据b的长度扩充行。
把数组b当作一列,再根据a的长度扩充列。
torch中meshgrid:
把数组a当作一列,再根据b的长度扩充列。
把数组b当作一行,再根据a的长度扩充行。
torch.meshgrid
中有个参数indexing
可以调整顺序,默认是'ij'
'ij'
:先列后行
'xy'
:先行后列
numpy中meshgrid:
把数组a当作一行,再根据b的长度扩充行。
把数组b当作一列,再根据a的长度扩充列。
torch中meshgrid:
把数组a当作一列,再根据b的长度扩充列。
把数组b当作一行,再根据a的长度扩充行。
torch.meshgrid
中有个参数indexing
可以调整顺序,默认是'ij'
'ij'
:先列后行
'xy'
:先行后列