代码之家  ›  专栏  ›  技术社区  ›  user3668129

如何创建具有默认值的形状(1,1,32)的torch.ttensor?

  •  0
  • user3668129  · 技术社区  · 3 年前

    我想创建 torch.tensor 具有默认值的形状变量(1,1,32)( None ).

    我该怎么做?

    0 回复  |  直到 3 年前
        1
  •  2
  •   Ivan Jeffrey Zhao    3 年前

    我不相信你能分配 None torch.Tensor 。然而,更合适的做法是分配 NaN 。您可以使用内置 torch.full :

    >>> torch.full((1, 1, 32), torch.nan)
    tensor([[[nan, nan, nan, nan, nan, nan, nan, nan, 
              nan, nan, nan, nan, nan, nan, nan, nan, 
              nan, nan, nan, nan, nan, nan, nan, nan, 
              nan, nan, nan, nan, nan, nan, nan, nan]]])