For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Python function

local_dim_at

local_dim_at()

max.tensor.sharding.per_shard_dim.local_dim_at(d, shard)

source

Returns the dimension value at shard.

If d is a regular (non-distributed) Dim, then d is returned unchanged.

Parameters:

• d (Dim) – The dim to read.
• shard (int) – The device index within the mesh.

Raises:

IndexError – If d is a PerShardDim and shard is not a valid index into its cells.

Return type:

Dim