python中numpy转换成张量
时间: 2023-10-11 15:07:58 浏览: 94
使用python Numpy
要在Python中将numpy数组转换为张量,可以使用PyTorch中的torch.from_numpy()函数。这个函数将接受一个numpy数组作为输入并返回一个对应的张量对象。以下是一个示例代码:
import torch
import numpy as np
# 创建一个numpy数组
a = np.array([1, 2, 3])
# 将numpy数组转换为张量
t = torch.from_numpy(a)
# 打印结果
print(t)
在这个例子中,我们首先创建了一个numpy数组a,然后使用torch.from_numpy()函数将其转换为一个张量t。最后,我们打印出t的值。
相应的输出将是:
tensor([1, 2, 3])
阅读全文