深度学习 - 框架之 pytorch

import torch
import numpy as np
a = torch.Tensor([1, 2, 3])
b = torch.Tensor([4, 5, 6])

a_x, b_y = torch.meshgrid(a, b)
print("torch.meshgrid: ")
print(a_x)
print(b_y)

n_a = a.numpy()
n_b = b.numpy()

n_a_x, n_b_y = np.meshgrid(n_a, n_b)
print("========================\nnumpy meshgrid:")
print(n_a_x)

print(n_b_y)


torch.meshgrid: 
tensor([[1., 1., 1.],
        [2., 2., 2.],
        [3., 3., 3.]])
tensor([[4., 5., 6.],
        [4., 5., 6.],
        [4., 5., 6.]])
========================
numpy meshgrid:
[[1. 2. 3.]
 [1. 2. 3.]
 [1. 2. 3.]]
[[4. 4. 4.]
 [5. 5. 5.]
 [6. 6. 6.]]