Transpose

class Transpose(axes: IntArray) : Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>
Content copied to clipboard

Reverse or permute the axes of an input tensor.

Constructors

Transpose
Link copied to clipboard
fun Transpose(axes: IntArray = intArrayOf(2, 0, 1))
Content copied to clipboard

Functions

apply
Link copied to clipboard
open override fun apply(input: Pair<FloatArray, TensorShape>): Pair<FloatArray, TensorShape>
Content copied to clipboard
getOutputShape
Link copied to clipboard
open override fun getOutputShape(inputShape: TensorShape): TensorShape
Content copied to clipboard

Properties

axes
Link copied to clipboard
var axes: IntArray
Content copied to clipboard

Array of ints, default value is related to the typical transpose task for H, W, C to C, W, H tensor format conversion.