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)
pytorch之tensor类型间的相互转换
最新推荐文章于 2023-02-15 23:42:03 发布