numpy转torch

将一个numpy数组转换为PyTorch张量可以使用torch.from_numpy()函数。

例如，下面的代码将一个numpy数组转换为PyTorch张量：

import numpy as np
import torch

# Create a numpy array
np_array = np.array([[1, 2, 3], [4, 5, 6]])

# Convert the numpy array to a PyTorch tensor
torch_tensor = torch.from_numpy(np_array)

print(torch_tensor)

输出：

tensor([[1, 2, 3],
        [4, 5, 6]])