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 multidimensional 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)