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_shape_at

local_shape_at()

max.experimental.sharding.per_shard_dim.local_shape_at(shape, shard)

source

Projects every axis of shape onto a single shard index.

Parameters:

• shape (Sequence[Dim])
• shard (int)

Return type:

Shape