python如何把线性数据转化为tensor
时间: 2023-10-05 22:09:52 浏览: 130
Pytorch中Tensor与各种图像格式的相互转化详解
5星 · 资源好评率100%
在 Python 中,可以使用 PyTorch 或 TensorFlow 等深度学习框架将线性数据(如列表、数组等)转换为 tensor。
以 PyTorch 为例,可以使用以下代码将列表转换为 tensor:
```python
import torch
data = [1, 2, 3, 4, 5]
tensor_data = torch.tensor(data)
print(tensor_data)
```
输出结果为:
```
tensor([1, 2, 3, 4, 5])
```
如果要将多维数据(如二维数组)转换为 tensor,可以使用 `torch.tensor()` 或 `torch.from_numpy()` 函数。例如:
```python
import numpy as np
import torch
data = np.array([[1, 2], [3, 4]])
tensor_data1 = torch.tensor(data)
tensor_data2 = torch.from_numpy(data)
print(tensor_data1)
print(tensor_data2)
```
输出结果为:
```
tensor([[1, 2],
[3, 4]])
tensor([[1, 2],
[3, 4]])
```
注意,使用 `torch.tensor()` 函数会复制数据,而使用 `torch.from_numpy()` 函数则会共享数据,因此如果修改 `tensor_data2`,会同时修改原始数据 `data`。
阅读全文