torch转numpy

要将 PyTorch 张量（Tensor）转换为 NumPy 数组，可以使用 `.numpy()` 方法。下面是一个示例代码：

import torch
import numpy as np

# 创建一个 PyTorch 张量
tensor = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 将张量转换为 NumPy 数组
numpy_array = tensor.numpy()

print(numpy_array)

输出：

[[1 2 3]
 [4 5 6]]

需要注意的是，将 PyTorch 张量转换为 NumPy 数组时，如果原始张量在 CPU 上，则转换后的数组也将在 CPU 上。如果原始张量在 GPU 上，则需要先将其移到 CPU 上，然后再进行转换。