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

Types

TensorShape
Link copied to clipboard
class TensorShape
Content copied to clipboard

Helper wrapper of Shape class with helper methods.

Functions

flattenFloats
Link copied to clipboard
fun Array<*>.flattenFloats(): FloatArray
Content copied to clipboard

Flattens the given array of float values.

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

Returns last dimensions (except first) from shape.