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

Mojo function

tile_shape

def tile_shape[input_type: DType, repeats_type: DType](input_buf: TileTensor[input_type, Storage=input_buf.Storage, address_space=input_buf.address_space, linear_idx_type=input_buf.linear_idx_type, element_size=input_buf.element_size], repeats_buf: TileTensor[repeats_type, Storage=repeats_buf.Storage, address_space=repeats_buf.address_space, linear_idx_type=repeats_buf.linear_idx_type, element_size=repeats_buf.element_size]) -> IndexList[input_buf.LayoutType.rank]

Compute the output shape of a tile operation, and assert the inputs are compatible.

Parameters:

input_type (DType): Type of the input tensor.
repeats_type (DType): Type of the repeats tensor.

Args:

input_buf (TileTensor[input_type, Storage=input_buf.Storage, address_space=input_buf.address_space, linear_idx_type=input_buf.linear_idx_type, element_size=input_buf.element_size]): The input tensor.
repeats_buf (TileTensor[repeats_type, Storage=repeats_buf.Storage, address_space=repeats_buf.address_space, linear_idx_type=repeats_buf.linear_idx_type, element_size=repeats_buf.element_size]): The repeats tensor.

Returns:

IndexList[input_buf.LayoutType.rank]: The output shape.