扫二维码与项目经理沟通
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
Summary
主要包括以下三种途径：
使用独立的函数；
使用torch.type()函数；
使用type_as(tesnor)将张量转换为给定类型的张量。
使用独立函数
import torch tensor = torch.randn(3, 5) print(tensor) # torch.long() 将tensor投射为long类型 long_tensor = tensor.long() print(long_tensor) # torch.half()将tensor投射为半精度浮点类型 half_tensor = tensor.half() print(half_tensor) # torch.int()将该tensor投射为int类型 int_tensor = tensor.int() print(int_tensor) # torch.double()将该tensor投射为double类型 double_tensor = tensor.double() print(double_tensor) # torch.float()将该tensor投射为float类型 float_tensor = tensor.float() print(float_tensor) # torch.char()将该tensor投射为char类型 char_tensor = tensor.char() print(char_tensor) # torch.byte()将该tensor投射为byte类型 byte_tensor = tensor.byte() print(byte_tensor) # torch.short()将该tensor投射为short类型 short_tensor = tensor.short() print(short_tensor)
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
微信二维码
移动版官网