torch.Tensor和torch.tensor有什么区别

原创

已于 2024-10-17 15:22:09 修改 · 231 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #人工智能

于 2024-01-31 00:04:04 首次发布

本文详细介绍了在PyTorch中如何创建和操作torch.Tensor对象，包括不同构造方式下的shape属性，并讨论了requires_grad属性的使用和限制。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在PyTorch 中，torch.Tensor是主要的tensor类，所有的tensor都是torch.Tensor的实例，但是torch.Tensor并没有requires_grad的参数。torch.tensor是torch.FloatTensor的别名。
最主要的是code上有一些区别

import torch

if __name__ == '__main__':

    # print(torch.Tensor([2.0]).shape) #torch.Size([1])
    # print(torch.tensor([2.0]).shape) #torch.Size([1])
    # # print(torch.Tensor(2.0).shape) #直接崩了
    # print(torch.tensor(2.0).shape) #torch.Size([])
    # print(torch.tensor(2.0)) #tensor(2.)

    print(torch.Tensor