python 中的torch.tensor
时间: 2024-02-25 09:30:13 浏览: 51
Python库 | torch-tensor-type-0.0.1.tar.gz
torch.tensor是PyTorch中的一个类,用于创建张量(n维数组)。它可以接受各种不同的输入类型(列表、元组、数组等)并返回一个张量。例如:
```
import torch
# 创建一个 3x3 的张量,元素都是0
x = torch.tensor([[0, 0, 0], [0, 0, 0], [0, 0, 0]])
# 创建一个 2x2 的张量,元素是随机数
y = torch.tensor([[0.1, 0.2], [0.3, 0.4]])
```
可以通过类似Numpy的方式进行张量的操作,例如索引、切片、数学运算等。同时,PyTorch还支持GPU加速,可以使用.to()方法将张量转移到GPU上进行运算。
阅读全文