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

Functions

head
Link copied to clipboard
fun Shape.head(): Long
Content copied to clipboard

Returns first dimension

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

Returns last dimensions (except first).

toShape
Link copied to clipboard
fun TensorShape.toShape(): Shape
Content copied to clipboard

Converts TensorShape to Shape object.

toTensorShape
Link copied to clipboard
fun Shape.toTensorShape(): TensorShape
Content copied to clipboard

Converts Shape to TensorShape object.