tensor转numpy
时间: 2023-10-19 15:03:25 浏览: 67
tensor与numpy转换资源合集
5星 · 资源好评率100%
可以使用numpy()方法将Tensor转换为Numpy数组,例如:
```
import torch
import numpy as np
# 创建一个大小为(2, 2)的Tensor
x = torch.tensor([[1, 2], [3, 4]])
# 将Tensor转为Numpy数组
y = x.numpy()
print(y)
# 输出结果为:
# array([[1, 2],
# [3, 4]], dtype=int64)
```
阅读全文