import torch #获取张量的数据类型 torch.tensor([1.2,3.4]).dtype #设置张量的默认数据类型 torch.set_default_tenser_type(torch.DoubleTensor) #将张量转换为其他类型 a=torch.tensor([1.2,3.4]) a.long()#转换为int64 a.int()#转换为int32 a.float()#转换为float32 #获取默认数据类型 torch.get_default_dtype()
pytorch深度学习入门与实战1:张量的数据类型
于 2024-06-04 19:50:19 首次发布