padded(forSizes:with:)
Returns a tensor padded with the given constant according to the specified padding sizes.
Declaration
func padded(forSizes sizes: [(before: Int, after: Int)], with value: Float) -> MLTensorParameters
- sizes:
An array of tuples describing the size to be inserted before and after each dimension.
- value:
The constant value used for padding.
Return Value
The padded tensor.
Discussion
For example:
let x = MLTensor(shape: [2, 3], scalars: [
1, 2, 3,
4, 5, 6
], scalarType: Float32.self)
let y = x.padded(forSizes: [(0, 0), (2, 2)], with: 0.0)
// [[0, 0, 1, 2, 3, 0, 0],
// [0, 0, 4, 5, 6, 0, 0]]