Numpy与pytorch对比记忆

1 改变类型

numpy：array.astype()
torch: tensor.type()

import numpy as np
import torch

if __name__=="__main__":
    nn = np.array(1)
    tt = torch.tensor(1)

    nn.astype(dtype=np.long)
    nn.astype(dtype=np.int32)

    tt.float()
    tt.type(dtype=torch.float32)

参考：Numpy属性dtype的转换(数据类型转换):unit8和float32转换