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

Types

TensorShape
Link copied to clipboard
class TensorShape
Content copied to clipboard

Representation of the tensor shape class with helper methods.

Functions

getDimsOfArray
Link copied to clipboard
fun getDimsOfArray(data: Array<*>): LongArray
Content copied to clipboard

Get shape of array of arrays (of arrays...) of Array of elements of any type. If the most inner array does not have any elements its size is skipped in the result.

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

Wraps an IntArray to TensorShape.

Properties

tensorShape
Link copied to clipboard
val Array<*>.tensorShape: TensorShape
Content copied to clipboard