torch.tensor函数
时间: 2024-02-25 10:47:00 浏览: 85
Pytorch学习之torch用法—-比较操作(Comparison Ops)
torch.tensor函数是PyTorch库中用于创建张量的函数之一。它可以将一个序列(如列表)转换为一个张量,并且还可以根据需要指定数据类型和其他参数。使用torch.tensor函数可以方便地创建张量,而不需要显式地指定数据类型。该函数的示例如下所示:
```python
import torch
# 创建一个包含整型数据的张量
x = torch.tensor([1, 2, 3, 4, 5])
# 创建一个包含浮点型数据的张量
y = torch.tensor([0.1, 0.2, 0.3, 0.4, 0.5])
# 创建一个指定数据类型的张量
z = torch.tensor([1, 2, 3, 4, 5], dtype=torch.float32)
```
torch.tensor函数可以根据输入的数据自动推断张量的形状和数据类型,并返回一个新的张量对象。它是创建张量的基本方法之一,非常常用。
阅读全文