jax.ShapeDtypeStruct#

class jax.ShapeDtypeStruct(shape, dtype, *, sharding=None, weak_type=False)[原始碼]#

用於儲存陣列的形狀、dtype 和其他靜態屬性的容器。

ShapeDtypeStruct 經常與 jax.eval_shape() 結合使用。

參數:

  • shape – 代表陣列形狀的整數序列

  • dtype – 類似 dtype 的物件

  • sharding – (選填) jax.Sharding 物件

__init__(shape, dtype, *, sharding=None, weak_type=False)[原始碼]#

方法

__init__(shape, dtype, *[, sharding, weak_type])

屬性

shape

dtype

sharding

weak_type

layout

ndim

size