将数据转换为 PyTorch 张量
时间: 2024-05-16 08:16:24 浏览: 140
pytorch中tensor张量数据类型的转化方式
5星 · 资源好评率100%
可以使用 PyTorch 的 `torch.Tensor()` 函数将数据转换为张量。例如,将一个列表转换为张量的代码如下:
```python
import torch
data = [1, 2, 3, 4, 5]
tensor_data = torch.Tensor(data)
print(tensor_data)
```
输出结果为:
```
tensor([1., 2., 3., 4., 5.])
```
如果数据是多维的,可以使用 `torch.Tensor()` 或者 `torch.from_numpy()` 函数进行转换。例如,将一个二维的 NumPy 数组转换为张量的代码如下:
```python
import numpy as np
import torch
data = np.array([[1, 2, 3], [4, 5, 6]])
tensor_data = torch.Tensor(data)
print(tensor_data)
```
输出结果为:
```
tensor([[1., 2., 3.],
[4., 5., 6.]])
```
阅读全文