pytorch中numpy如何转换为tensor格式

在PyTorch中，我们可以使用torch.from_numpy()函数将numpy数组转换为tensor格式。具体步骤如下所示：

import numpy as np
import torch

# 创建一个numpy数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 将numpy数组转换为tensor格式
tensor = torch.from_numpy(arr)

print(tensor)
# 输出结果：tensor([[1, 2, 3],
#                  [4, 5, 6]], dtype=torch.int32)

需要注意的是，torch.from_numpy()函数返回的tensor和原始的numpy数组共享内存，因此对其中一个的修改也会影响到另一个。如果不希望共享内存，可以使用tensor.clone()函数创建一个新的tensor，例如：

tensor2 = tensor.clone()