torch.tensor
时间: 2023-09-12 18:03:26 浏览: 80
Python库 | torch-tensor-type-0.0.1.tar.gz
torch.tensor是PyTorch中用于创建张量(tensors)的函数。张量是一种多维数组,类似于NumPy中的数组,但是可以在GPU上进行计算。例如,可以使用以下代码创建一个形状为(3, 2)的张量:
```
import torch
tensor = torch.tensor([[1, 2], [3, 4], [5, 6]])
print(tensor)
```
输出:
```
tensor([[1, 2],
[3, 4],
[5, 6]])
```
可以看到,这个张量包含3行2列的整数。我们还可以通过指定dtype(数据类型)来创建具有不同数据类型的张量,例如:
```
tensor = torch.tensor([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]], dtype=torch.float)
print(tensor)
```
输出:
```
tensor([[1., 2.],
[3., 4.],
[5., 6.]])
```
这个张量包含3行2列的浮点数。可以使用各种算术和逻辑操作对张量进行操作,例如加法、减法、乘法、除法、转置等。
阅读全文