tensor转array

在PyTorch中，可以使用`numpy()`方法将Tensor转换为NumPy数组。以下是一个例子：

import torch

# 创建一个Tensor
tensor = torch.randn(3, 3)

# 将Tensor转换为NumPy数组
array = tensor.numpy()

print("Tensor:\n", tensor)
print("Array:\n", array)

输出结果：

Tensor:
 tensor([[-0.6506, -0.3575, -1.5589],
         [ 0.4260, -1.2344, -0.2654],
         [ 1.3034, -0.3057, -1.1795]])
Array:
 [[-0.65064085 -0.35750654 -1.5589143 ]
 [ 0.4259739  -1.2343692  -0.26540878]
 [ 1.3033649  -0.30574968 -1.179456  ]]

注意：当Tensor存储在GPU中时，需要先将其移动到CPU上，然后才能使用`numpy()`方法将其转换为NumPy数组。可以使用`to()`方法将Tensor从GPU移动到CPU。例如：

import torch

# 创建一个在GPU上的Tensor
tensor = torch.randn(3, 3).cuda()

# 将Tensor移动到CPU上，并转换为NumPy数组
array = tensor.to('cpu').numpy()

print("Tensor:\n", tensor)
print("Array:\n", array)