Package org.jetbrains.kotlinx.dl.api.core.shape

Types

TensorShape
Link copied to clipboard
class TensorShape

Helper wrapper of Shape class with helper methods.

Functions

reshape2DTo1D
Link copied to clipboard
fun reshape2DTo1D(dst: Array<FloatArray>, size: Int): FloatArray

Reshapes 2D array of floats to 1D array of floats.

reshape3DTo1D
Link copied to clipboard
fun reshape3DTo1D(dst: Array<Array<FloatArray>>, size: Int): FloatArray

Reshapes 3D array of floats to 1D array of floats.

reshape4DTo1D
Link copied to clipboard
fun reshape4DTo1D(dst: Array<Array<Array<FloatArray>>>, size: Int): FloatArray

Reshapes 4D array of floats to 1D array of floats.

tail
Link copied to clipboard
fun tail(shape: Shape): LongArray

Returns last dimensions (except first) from shape.