Python 中的 astype() 和 .dtype

这里主要介绍 astype(), .dtype看以下就能懂(下面包含 .dtype)

• .dtype 就是显示数据类型的一种方法

astype() 的作用

• 转换 numpy 数组的数据类型, 因此如果数据是 tensor 类型的数据, 就需要先将数据转换为 numpy.array 类型再进行转换

示例

import torch
import numpy as np

a = torch.rand(2, 2)
print(a.numpy().astype('uint8').dtype)
print(a)
print(a.numpy().astype('float32').dtype)
print(a)

>>uint8
>>tensor([[0.4651, 0.8008],
        [0.8621, 0.3698]])
>>float32
>>tensor([[0.4651, 0.8008],
        [0.8621, 0.3698]])

注意

在使用 astype() 函数时, 想要变换的类型必须加引号