【Pytorch】PyTorch: meshgrid的用法(与tensorflow的差异)

meshgrid

Example::>>>x=torch.tensor([1,2,3])>>>y=torch.tensor([4,5,6])>>>grid_x,grid_y=torch.meshgrid(x,y)>>>grid_x tensor([[1,1,1],[2,2,2],[3,3,3]])>>>grid_y tensor([[4,5,6],[4,5,6],[4,5,6]])

tensorflow:

x=np.array([1,2,3])y=np.array([4,5,6])grid_x,grid_y=tf.meshgrid(x,y)sess=tf.Session()print(sess.run(grid_x))print(sess.run(grid_y))results:[[123][123][123]][[444][555][666]]

做代码重构的时候发现tensorflow和pytorch的meshgrid的实现是不一致的,要保持一致需要后续进行transpose和reshape操作,以此简单记录.