jax.numpy.dtype#

class jax.numpy.dtype(dtype, align=False, copy=False[, metadata])#

创建一个数据类型对象。

一个 numpy 数组是同质的,并且包含由 dtype 对象描述的元素。一个 dtype 对象可以由基本数值类型的不同组合构建。

参数
  • dtype – 要转换为数据类型对象的对象。

  • align (bool, 可选) – 向字段添加填充,以匹配 C 编译器为类似的 C 结构输出的内容。只有当 obj 是字典或逗号分隔的字符串时,才能为 True。如果正在创建结构化 dtype,这也会设置一个粘性对齐标志 isalignedstruct

  • copy (bool, 可选) – 创建数据类型对象的新副本。如果 False,结果可能只是对内置数据类型对象的引用。

  • metadata (dict, 可选) – 包含 dtype 元数据的可选字典。

另请参阅

result_type

示例

使用数组标量类型

>>> import numpy as np
>>> np.dtype(np.int16)
dtype('int16')

结构化类型,一个字段名为 'f1',包含 int16

>>> np.dtype([('f1', np.int16)])
dtype([('f1', '<i2')])

结构化类型,一个字段名为 'f1',其本身包含一个带有一个字段的结构化类型

>>> np.dtype([('f1', [('f1', np.int16)])])
dtype([('f1', [('f1', '<i2')])])

结构化类型,两个字段:第一个字段包含一个无符号整数,第二个字段包含一个 int32

>>> np.dtype([('f1', np.uint64), ('f2', np.int32)])
dtype([('f1', '<u8'), ('f2', '<i4')])

使用数组协议类型字符串

>>> np.dtype([('a','f8'),('b','S10')])
dtype([('a', '<f8'), ('b', 'S10')])

使用逗号分隔的字段格式。形状为 (2,3)

>>> np.dtype("i4, (2,3)f8")
dtype([('f0', '<i4'), ('f1', '<f8', (2, 3))])

使用元组。int 是一个固定类型,3 是字段的形状。void 是一个灵活类型,此处大小为 10

>>> np.dtype([('hello',(np.int64,3)),('world',np.void,10)])
dtype([('hello', '<i8', (3,)), ('world', 'V10')])

int16 细分为 2 个 int8,分别称为 x 和 y。0 和 1 是以字节为单位的偏移量

>>> np.dtype((np.int16, {'x':(np.int8,0), 'y':(np.int8,1)}))
dtype((numpy.int16, [('x', 'i1'), ('y', 'i1')]))

使用字典。两个字段分别名为 'gender' 和 'age'

>>> np.dtype({'names':['gender','age'], 'formats':['S1',np.uint8]})
dtype([('gender', 'S1'), ('age', 'u1')])

以字节为单位的偏移量,此处为 0 和 25

>>> np.dtype({'surname':('S25',0),'age':(np.uint8,25)})
dtype([('surname', 'S25'), ('age', 'u1')])
__init__()#

方法

__init__()

newbyteorder([new_order])

返回具有不同字节顺序的新 dtype。

属性

alignment

根据编译器,此数据类型所需的对齐方式(以字节为单位)。

base

返回子数组的基本元素的 dtype,无论其维度或形状如何。

byteorder

一个字符,指示此数据类型对象的字节顺序。

char

每个 21 种不同内置类型的唯一字符代码。

descr

数据类型的 __array_interface__ 描述。

fields

为此数据类型定义的命名字段的字典,或者 None

flags

描述如何解释此数据类型的位标志。

hasobject

一个布尔值,指示此 dtype 是否在任何字段或子 dtype 中包含任何引用计数的对象。

isalignedstruct

一个布尔值,指示 dtype 是否是保持字段对齐的结构。

isbuiltin

一个整数,指示此 dtype 与内置 dtype 的关系。

isnative

一个布尔值,指示此 dtype 的字节顺序是否是平台的原生顺序。

itemsize

此数据类型对象的元素大小。

kind

一个字符代码('biufcmMOSUV' 之一),标识数据的一般类型。

metadata

要么是 None,要么是元数据的只读字典(mappingproxy)。

name

此数据类型的位宽名称。

names

字段名称的有序列表,如果没有字段,则为 None

ndim

如果此数据类型描述子数组,则为子数组的维数,否则为 0

num

每个 21 种不同内置类型的唯一数字。

shape

如果此数据类型描述子数组,则为子数组的形状元组,否则为 ()

str

此数据类型对象的数组协议类型字符串。

subdtype

如果此 dtype 描述子数组,则为元组 (item_dtype, shape),否则为 None。

type