代码之家  ›  专栏  ›  技术社区  ›  Tom Hale

获取PyTorch张量的数据类型

  •  0
  • Tom Hale  · 技术社区  · 7 年前

    如何找出PyTorch张量中元素的类型?

    1 回复  |  直到 7 年前
        1
  •  16
  •   prosti    6 年前

    dtype                   || CPU tensor               || GPU tensor
    
    torch.float32              torch.FloatTensor           torch.cuda.FloatTensor
    

    你得到的第一个 print(t.dtype) 如果 t 是你的张量,否则你用 t.type() 另外两个。

        2
  •  7
  •   Tom Hale    7 年前

    dtype attribute :

    >>> import torch
    >>> print(torch.Tensor(1).dtype)
    torch.float32
    

    文档还提供了 canonical list of datatypes .