Skip to main content

Mojo function

concat

concat[input_origin: ImmutOrigin, InputLayoutType: TensorLayout, //, dtype: DType, single_thread_blocking_override: Bool, target: StringSlice[StaticConstantOrigin] = StringSlice("cpu"), epilogue_fn: Optional[def[c_type: DType, rank: Int, width: Int = 1, *, alignment: Int = 1](IndexList[rank], SIMD[c_type, width]) capturing -> None] = None](output: TileTensor[dtype, output.LayoutType, output.origin, linear_idx_type=output.linear_idx_type, element_size=output.element_size], axis: Int, inputs: StaticTuple[TileTensor[dtype, InputLayoutType, input_origin], inputs.size], context: DeviceContextPtr = DeviceContextPtr())

Was this page helpful?