Mojo function
pad_reflect
pad_reflect[rank: Int, output_shape: DimList, input_shape: DimList, type: DType, paddings_type: DType](output: NDBuffer[type, rank, origin, output_shape], input: NDBuffer[type, rank, origin, input_shape], paddings: UnsafePointer[SIMD[paddings_type, 1]])
Fill output
with values from input
, and edges padded with reflected values from the unpadded region.
Example: var input = [[1, 2], [3, 4]] var paddings = [2, 2, 1, 0]
Yields: output = [[2, 1, 2], [4, 3, 4], [2, 1, 2], [4, 3, 4], [2, 1, 2], [4, 3, 4]]
Args:
- output (
NDBuffer[type, rank, origin, output_shape]
): The output buffer. - input (
NDBuffer[type, rank, origin, input_shape]
): The input buffer. - paddings (
UnsafePointer[SIMD[paddings_type, 1]]
): Ordered (before, after) padding sizes for each axis.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!