怎么将数据转为tensor形
时间: 2023-05-20 18:02:40 浏览: 128
PyTorch中Tensor的数据统计示例
式?
将数据转为tensor形式可以使用PyTorch中的torch.Tensor()函数。该函数可以将numpy数组、Python列表或元组等数据类型转换为tensor形式。例如,以下代码将一个numpy数组转换为tensor形式:
import torch
import numpy as np
data = np.array([[1, 2], [3, 4]])
tensor_data = torch.Tensor(data)
print(tensor_data)
输出结果为:
tensor([[1., 2.],
[3., 4.]])
这样就将数据成功转换为tensor形式了。
阅读全文