2024/6/25 学习笔记 numpy数据内部类型转换方法

常见的numpy数据类型有：
np.int8、np.int16、np.int32、np.int64、
np.uint8、np.uint16、np.uint32、np.uint64、
np.float16、np.float32、np.float64

---

1.创建指定类型的numpy数据：

import numpy as np
#1.通过dtype参数指定（重要）:
#torch也可以通过指定dtype参数来指定tensor类型
b=np.ones((3,3),dtype=np.int16)

#2.通过类似np.int16（）、np.float64()、np.uintment()等方法指定:
#与torch.ShortTensor()、torch.DoubleTensor()、torch.ByteTensor()等方法类似
b=np.int16([3,2])
print(b.dtype)

---

2.创建好的numpy数据，改变其类型：

import numpy as np

#1.通过numpy数据的astype()方法指定（重要）:
#与tensor数据的type()方法类似:
b=b.astype(np.float32)
print(b.dtype)

#2.通过类似np.int16()、np.float32()等方法进行包装
#与torch.ShortTensor()、torch.DoubleTensor()、torch.ByteTensor()等方法类似
b=np.int16(b)
print(b.dtype)

#3.通过np.array()和dtype参数进行包装:
#类似torch.Tensor()与dtype参数结合使用
np.array(b,dtype=np.int16)

---