这是创建 PyTorch 标量的方法吗?

Sim*_*ity 4 python pytorch

我是 PyTorch 的新手,只是想确认以下内容是否分别创建了值为 1、2 和 3 的标量?

import torch

a = torch.tensor(1)
b = torch.tensor(2)
c = torch.tensor(3)
Run Code Online (Sandbox Code Playgroud)

谢谢。

Dai*_*air 7

来自文档中的示例torch.tensor

>>> torch.tensor(3.14159)  # Create a scalar (zero-dimensional tensor)
tensor(3.1416)
Run Code Online (Sandbox Code Playgroud)

因此,他们似乎赞同传递一个数字会创建相应的标量。

  • @MohitLamba 一个用例:如果您想使用 PyTorch 的 JIT 跟踪功能,您的函数只能采用“torch.Tensor”类型参数(及其元组、列表和字典)。因此,您的函数需要读取“my_function(f: torch.Tensor, t: torch.Tensor)”,而不是“my_function(f: float, t: torch.Tensor)”。 (2认同)