Python 中的 astype() 和 .dtype
Posted 怎样才能回到过去
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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() 函数时, 想要变换的类型必须加引号
以上是关于Python 中的 astype() 和 .dtype的主要内容,如果未能解决你的问题,请参考以下文章