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

原创

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

0 ·

CC 4.0 BY-SA版权

文章标签：

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

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

在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