torch.tensor()与torch.Tensor()的区别

我已经流浪了如此之久

于 2022-10-29 14:13:18 发布

阅读量650

点赞数 2

分类专栏： # pytorch 文章标签： python pytorch

本文链接：https://blog.csdn.net/gehueuwhgw/article/details/127586220

版权

3 篇文章 0 订阅

订阅专栏

torch.tensor与torch.Tensor的区别

torch.Tensor()的默认数据类型为torch.float32，而torch.tensor()的数据类型为原类型。
torch.Tensor()传入数字表示的形状与torch.FloatTensor相同
```
a = torch.Tensor(1)
print(a.type())
```
- 当torch.Tensor()传入的是一个数字时，返回的不是一个tensor类型的数字，而是一个tensor类型的列表，并且此列表的长度等于传入的数字，列表中的数据类型默认为float32。
```
b = torch.Tensor(3)
print(b)
```
- 当torch.tensor()传入的是一个数字时，返回的是tensor类型的数字
```
c = torch.tensor(3)
print(c)
```
torch.Tensor传入可迭代对象，返回值为tensor类型的列表，且里面的数据默认为float32
```
d = torch.Tensor([1, 2, 3])
print(d)
```
torch.tensor传入可迭代对象，返回值为tensor类型的列表，且里面的数据的数据类型为输入时可迭代对象里面的数据类型。
```
e = torch.tensor([2, 2, 3])
print(e)
```