Transpose

class Transpose(axes: IntArray) : Preprocessor
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(data: FloatArray, inputShape: ImageShape): FloatArray
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.