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).

Mojo function

scatter_min_shape_fn

def scatter_min_shape_fn(input: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec], updates: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=updates.static_spec], indices: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=indices.static_spec], axis: Scalar) -> IndexList[input.static_spec.rank]

Computes the output shape for the mo.scatter.min graph op.

Returns:

IndexList[input.static_spec.rank]