PyTorch | Tensor 和 tensor 的区别

PyTorch | Tensor 和 tensor 的区别

1. 介绍

\quad P y T o r c h PyTorch PyTorch 中,torch.Tensor是一种主要的 t e n s o r tensor tensor 类型,是torch.FloatTensor()的别名。所有的 t e n s o r tensor tensor 都是torch.Tensor的实例。
\quad torch.tensor()是一个函数,函数原型是:

torch.tensor(data, dtype=None, device=None, requires_grad=False)

\quad 其中 d a t a data data 可以是:listtupleNumPy ndarrayscalar和其他类型。
\quad torch.tensor会从 d a t a data data 中的数据部分做拷贝(而不是直接引用),根据原始数据类型生成相应的torch.LongTensortorch.FloatTensortorch.DoubleTensor或者根据 d t y p e dtype dtype 的值生成相应数据类型。

2. 区别

  • t o r c h . T e n s o r ( d a t a ) torch.Tensor(data) torch.Tensor(data):将输入的data转化torch.FloatTensor()
  • t o r c h . t e n s o r ( d a t a ) torch.tensor(data) torch.tensor(data):将data转化为torch.FloatTensortorch.LongTensortorch.DoubleTensor等类型,转化类型依据于data的类型或者dtype的值。
  • t o r c h . T e n s o r ( ) torch.Tensor() torch.Tensor() 可以创建一个空的 F l o a t T e n s o r FloatTensor FloatTensor,使用 t o r c h . t e n s o r ( ) torch.tensor() torch.tensor() 时则会报错。
    在这里插入图片描述
    \quad 当你想要创建一个空的tensor时候,可以使用如下的方法:
    \quad\quad torch.Tensor() # tensor([])
    \quad\quad torch.tensor(()) # tensor([])
    \quad\quad torch.empty([]) # tensor(0.)

猜你喜欢

转载自blog.csdn.net/m0_52650517/article/details/120456732