如何将numpy转化为pytorch张量
时间: 2023-03-28 11:01:14 浏览: 151
Pytorch使用PIL和Numpy将单张图片转为Pytorch张量方式
可以使用torch.from_numpy()函数将numpy数组转化为pytorch张量。例如,如果numpy数组为arr,则可以使用以下代码将其转化为pytorch张量:
import torch
import numpy as np
arr = np.array([1, 2, 3])
tensor = torch.from_numpy(arr)
注意,numpy数组和pytorch张量共享相同的内存,因此对numpy数组的修改也会影响到pytorch张量。
阅读全文