Package org.jetbrains.kotlinx.dl.api.extension

Functions

argmax

Returns the index of the maximum element in the given FloatArray. TODO: Should be replaced with Multik in future.

convertTensorToFlattenFloatArray

Copies tensor data to float array.

convertTensorToMultiDimArray

Copies tensor to multi-dimensional float array. Array rank is equal to tensor rank.

get2D

fun FloatArray.get2D(rowIndex: Int, columnIndex: Int, width: Int): Float

get3D

fun FloatArray.get3D(rowIndex: Int, columnIndex: Int, channelIndex: Int, width: Int, channels: Int): Float

set2D

fun FloatArray.set2D(rowIndex: Int, columnIndex: Int, width: Int, value: Float)

set3D

fun FloatArray.set3D(rowIndex: Int, columnIndex: Int, channelIndex: Int, width: Int, channels: Int, value: Float)