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

taoqick

已于 2024-02-02 00:09:49 修改

阅读量152

点赞数 1

文章标签： pytorch 深度学习人工智能

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

本文链接：https://blog.csdn.net/taoqick/article/details/135943835

版权

在PyTorch 中，torch.Tensor是主要的tensor类，所有的tensor都是torch.Tensor的实例。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(2).shape) #torch.Size([2])
    print(torch.tensor(2).shape) #torch.Size([])
    print(torch.empty(2)) #tensor([0., 0.])
    print(torch.empty(0).shape) #torch.Size([0])
    print(torch.tensor(2, requires_grad=True).shape) #torch.Size([])
	#print(torch.tensor(2, requires_grad=True).shape) # 崩了，RuntimeError: Only Tensors of floating point and complex dtype can require gradients
	print(torch.tensor(2.0, requires_grad=True).shape) # torch.Size([])
	print(torch.tensor([2.0], requires_grad=True).shape) # torch.Size([])
	#print(torch.Tensor([2.0], requires_grad=True).shape) # 也会崩，Tensor是类，得用小写的tensor函数才行