jax.Array.view

内容

jax.Array.view#

abstract Array.view(dtype=None, type=None)[source]#

返回数组的按位副本,将其视为新的 dtype。

这是围绕 jax.lax.bitcast_convert_type() 的功能更全面的包装器。

如果源 dtype 和目标 dtype 具有相同的位宽,则结果与输入数组具有相同的形状。如果目标 dtype 的位宽与源 dtype 不同,则结果的最后一个轴的大小将相应调整。

>>> jnp.zeros([1,2,3], dtype=jnp.int16).view(jnp.int8).shape
(1, 2, 6)
>>> jnp.zeros([1,2,4], dtype=jnp.int8).view(jnp.int16).shape
(1, 2, 2)

涉及布尔值的转换在所有情况下都不是很好定义的。关于上面解释的结果形状,布尔值被视为具有 8 位的位宽。但是,当转换为布尔值数组时,输入应只包含 0 或 1 字节。否则,结果可能不可预测,或者可能根据结果的使用方式而改变。

此转换是保证的且安全的

>>> jnp.array([1, 0, 1], dtype=jnp.int8).view(jnp.bool_)
Array([ True, False,  True], dtype=bool)

但是,对于涉及此类视图的任何表达式的结果,没有任何保证,例如:jnp.array([1, 2, 3], dtype=jnp.int8).view(jnp.bool_)。特别是,结果可能会在 JAX 版本之间以及根据平台的不同而发生变化。要安全地将此类数组转换为布尔数组,请将其与0进行比较。

>>> jnp.array([1, 2, 0], dtype=jnp.int8) != 0
Array([ True,  True, False], dtype=bool)
参数:
  • self (Array)

  • dtype (DTypeLike | None)

  • type (None)

返回类型:

Array