TensorShape

fun TensorShape(dims: LongArray)
Content copied to clipboard

Creates a new TensorShape with the given dimensions.

Parameters

dims

The sizes of the remaining dimensions

fun TensorShape(firstDimension: Long, vararg dims: Long)
Content copied to clipboard

Creates a new TensorShape with the given dimensions.

Parameters

firstDimension

The size of the first dimension

dims

The sizes of the remaining dimensions

fun TensorShape()
Content copied to clipboard