python 转tensor
时间: 2023-06-08 07:04:17 浏览: 105
将 Python 数据类型转换为 PyTorch 的 tensor 对象,可以使用 torch.Tensor() 函数,例如:
```python
import torch
# 转换一个 Python 列表为 tensor
my_list = [1, 2, 3, 4, 5]
my_tensor = torch.Tensor(my_list)
# 转换一个 Numpy 数组为 tensor
import numpy as np
my_array = np.array([1, 2, 3, 4, 5])
my_tensor = torch.Tensor(my_array)
```
注意,将 Python 的数据类型转换为 tensor 之后,可以在 GPU 中进行加速计算,提高程序的运行效率。
阅读全文