Skip to main content
Log in

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.