TensorShape

class TensorShape

Representation of the tensor shape class with helper methods.

Constructors

TensorShape

fun TensorShape(dims: LongArray)

Creates a new TensorShape with the given dimensions.

TensorShape

fun TensorShape(firstDimension: Long, vararg dims: Long)

Creates a new TensorShape with the given dimensions.

TensorShape

fun TensorShape()

Types

Companion

object Companion

Functions

almostEqual

fun almostEqual(tensorShape: TensorShape, except: Int): Boolean

Check the fact that two shapes has the same values at the same dimensions except one with index except.

assertKnown

fun assertKnown(i: Int)

Throw an exception if dimension i is unknown.

clone

fun clone(): TensorShape

Makes a copy of TensorShape object.

concatenate

fun concatenate(vararg dims: Long): TensorShape

Augment this TensorShape by appending more dimensions to it.

dims

fun dims(): LongArray

Returns the array of dimensions representing this shape.

equals

open operator override fun equals(other: Any?): Boolean

get

operator fun get(i: Int): Long

Returns the value of a dimension

hashCode

open override fun hashCode(): Int

head

fun head(): Long

Returns the head dimension.

numElements

fun numElements(): Long

Returns amount of elements in Tensor with the given shape.

rank

fun rank(): Int

Returns the rank of this shape.

replace

fun replace(i: Int, dim: Long): TensorShape

Replace dimension i with a new dimension size.

replaceFirst

fun replaceFirst(dim: Long): TensorShape

Replace the first dimension with a new dimension size.

replaceLast

fun replaceLast(dim: Long): TensorShape

Replace the last dimension with a new dimension size.

set

operator fun set(i: Int, value: Long)

Sets the value of a dimension

size

fun size(i: Int): Long

Get the size of a target dimension.

tail

fun tail(): LongArray

Returns the tail dimension.

toString

open override fun toString(): String