torch.meshgrid与np.meshgrid区别

之前一直认为torch中的meshgrid与numpy中的meshgrid是一个函数,今天在使用的时候踩坑了,终于发现原来输出的x,y坐标对换了 。

numpy.meshgrid函数
在这里插入图片描述

torch.meshgrid函数
在这里插入图片描述
主要区别在于生成的行列正好相反

猜你喜欢

转载自blog.csdn.net/caobin_cumt/article/details/129052506
今日推荐