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

Functions

convertTensorToFlattenFloatArray
Link copied to clipboard
fun Tensor<*>.convertTensorToFlattenFloatArray(): FloatArray

Copies tensor data to float array.

convertTensorToMultiDimArray
Link copied to clipboard
fun Tensor<*>.convertTensorToMultiDimArray(): Array<*>

Copies tensor to multidimensional float array. Array rank is equal to tensor rank.