Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Showing content from https://docs.jax.dev/en/latest/_autosummary/jax.ShapeDtypeStruct.html below:

jax.ShapeDtypeStruct — JAX documentation

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

A container for the shape, dtype, and other static attributes of an array.

ShapeDtypeStruct is often used in conjunction with jax.eval_shape().

Parameters:

• shape – a sequence of integers representing an array shape

• dtype – a dtype-like object

• sharding – (optional) a jax.Sharding object

__init__(shape, dtype, *, sharding=None, weak_type=False)[source]#

Methods

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

update(**kwargs)

Attributes

shape

dtype

sharding

weak_type

layout

ndim

size

RetroSearch is an open source project built by @garambo | Open a GitHub Issue

Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo

HTML: 3.2 | Encoding: UTF-8 | Version: 0.7.3