可以使用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数组的值也会发生相应的改变。

将np转化为tensor

原文地址: https://www.cveoy.top/t/topic/hQv1 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录