将np转化为tensor
可以使用torch.from_numpy()函数将numpy数组转换为tensor。以下是一个例子:
import numpy as np
import torch
# 创建一个numpy数组
np_array = np.array([1, 2, 3, 4, 5])
# 将numpy数组转换为tensor
tensor = torch.from_numpy(np_array)
print(tensor) # 输出: tensor([1, 2, 3, 4, 5])
注意,这种转换方式不会创建新的内存空间,而是共享内存。所以,如果修改了tensor的值,那么numpy数组的值也会发生相应的改变。
原文地址: https://www.cveoy.top/t/topic/hQv1 著作权归作者所有。请勿转载和采集!