Pytorch框架之tensor类型转换(type, type_as)

                         QQ:3020889729                                                                                 小蔡

type – 指定类型改变

原型：type(dtype=None, non_blocking=False, **kwargs)
按输入的数据类型进行转换并返回
【注意： 如果不指定数据类型，就返回本身的数据】

data = torch.ones(2, 2)
print(data.dtype)
#result: torch.int64
# 可能在操作过程中指定其他数据类型--这里就按照ones--对应int64类型
data = data.type(torch.float32)  # 要接收类型已经改变的tensor数据，否