pytorch 修改tensor数据类型

import torch

a = torch.randn(3, 200, 200)
print(a.dtype)

b = a.type(torch.float16)
print(b.dtype)

c = a.type(torch.int32)
print(c.dtype)

d = a.type(torch.long)
print(d.dtype)

e = a.type(torch.float32)
print(e.dtype)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16