Mojo function

pad_repeat

pad_repeat[output_layout: Layout, input_layout: Layout, dtype: DType, paddings_type: DType](output: LayoutTensor[dtype, output_layout, origin, address_space=address_space, element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment], input: LayoutTensor[dtype, input_layout, origin, address_space=address_space, element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment], paddings: UnsafePointer[Scalar[paddings_type]])

Fill output with values from input, and edges padded boundary values from the unpadded region.

Example: var input = [[1, 2], [3, 4]] var paddings = [2, 2, 1, 0]

Yields: output = [[1, 1, 2], [1, 1, 2], [1, 1, 2], [3, 3, 4], [3, 3, 4], [3, 3, 4]]

Parameters:

output_layout (Layout): Layout of the output buffer.
input_layout (Layout): Layout of the input buffer.
dtype (DType): DType of the input/output buffer.
paddings_type (DType): DType of the input, output, and padding buffers.

Args:

output (LayoutTensor): The output buffer.
input (LayoutTensor): The input buffer.
paddings (UnsafePointer): Ordered (before, after) padding sizes for each axis.