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

concat_from_list_shape_impl

def concat_from_list_shape_impl[dtype: DType, rank: Int](axis0: Int, inputs: List[ManagedTensorSlice[IOSpec[_, _].Input, static_spec=_get_unknown_tensor_spec[dtype, rank]()]]) -> IndexList[rank]

Computes the concatenated output shape for a variadic list of input tensors.

Parameters:

• ​dtype (DType): Element type of the input tensors.
• ​rank (Int): Number of dimensions in each input tensor.

Args:

• ​axis0 (Int): Axis along which to concatenate; negative values index from the last dimension.
• ​inputs (List[ManagedTensorSlice[IOSpec[_, _].Input, static_spec=_get_unknown_tensor_spec[dtype, rank]()]]): List of input tensors to concatenate, all matching in shape except along the concat axis.

Returns:

IndexList[rank]: The output shape with the concat axis dimension summed across all inputs.