Package org.jetbrains.kotlinx.dl.onnx.dataset.preprocessor
Types
ONNXModelPreprocessor
Link copied to clipboard
class ONNXModelPreprocessor(onnxModel: OnnxInferenceModel?, outputIndex: Int) : Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>
Content copied to clipboard
Applies the given onnxModel as a preprocessing stage.
Transpose
Link copied to clipboard
class Transpose(axes: IntArray) : Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>
Content copied to clipboard
Reverse or permute the axes of an input tensor.
Functions
onnx
Link copied to clipboard
fun <I> Operation<I, Pair<FloatArray, TensorShape>>.onnx(block: ONNXModelPreprocessor.() -> Unit): Operation<I, Pair<FloatArray, TensorShape>>
Content copied to clipboard
Image DSL Preprocessing extension.
transpose
Link copied to clipboard
fun <I> Operation<I, Pair<FloatArray, TensorShape>>.transpose(sharpBlock: Transpose.() -> Unit): Operation<I, Pair<FloatArray, TensorShape>>
Content copied to clipboard
The DSL extension function for Transpose operation.