numpy.ndarray转tensor
时间: 2023-05-01 15:01:07 浏览: 184
计算机视觉——【tensorflow入门】Tensor与Numpy.ndarray的相互转换 计算机视觉.pdf
可以使用torch.from_numpy()函数将numpy.ndarray转换为tensor。例如:
```
import numpy as np
import torch
a = np.array([1, 2, 3])
tensor_a = torch.from_numpy(a)
print(tensor_a)
```
输出结果为: tensor([1, 2, 3], dtype=torch.int32)
阅读全文