如何判断一个torch.tensor dtype是不是int？

Question

我想检查 Tensor 从图像转换是否标准化，即 dtype 是 int 或 float。有没有方便的方法来实现这个目标？我不想要像a.dtype == torch.int or a.dtype == torch.int32 or a.dtype ==torch.uint8 .... 这样的枚举条件检查。还是有另一种方法来检查图像张量是否归一化？

Answer 1

正如 , you can use torch.is_floating_point 指出的那样检查你的张量：

if torch.is_floating_point(my_tensor):
  # the tensor is "normalized"...
else:
  # the tensor is an integer and needs to be normalized...

如何判断一个torch.tensor dtype是不是int？

How to judge a torch.tensor dtype is int or not?

pytorch