api/org.diffkt.tracing/TracingTensor/IfThenElse

IfThenElse

open class IfThenElse(cond: TracingTensor, whenTrue: TracingTensor, whenFalse: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Functions

accept

open override fun <R> accept(v: TracingVisitor<R>): R

equals

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

floatEval

open override fun floatEval(vars: FloatArray): Float

get

open operator fun get(index: Int): DTensor

open operator fun get(vararg indices: Int): DTensor

hashCode

open override fun hashCode(): Int

toCodeString

open fun toCodeString(): String

toString

open override fun toString(): String

wrap

open override fun wrap(wrapper: Wrapper): DTensor

Wrapper around the tensor

Properties

cond

val cond: TracingTensor

derivativeID

open override val derivativeID: DerivativeID

Each derivative is assigned a unique DerivativeID

indices

open val indices: Iterator<IntArray>

An iterator over the indices for the tensor

isScalar

open val isScalar: Boolean

True if the tensor is a scalar.

operations

open override val operations: Operations

The operations available on a tensor.

precomputedHashCode

val precomputedHashCode: Int

primal

open override val primal: DTensor

primal points to the actual tensor

rank

open val rank: Int

The number of dimensions in the tensor's shape. rank 0 - Scalar rank 1 - 1D array or 1D tensor rank 2 - 2D matrix or 2D tensor rank 3 - 3D tensor ... rank N - ND tensor

shape

open override val shape: Shape

shape indicates the number of dimension of a tensor and the length of each dimension. If the shape of the tensor is 3x4x5 then the value of shape is Shape(3,4,5).

size

open val size: Int

The total number of elements of this tensor.

whenFalse

val whenFalse: TracingTensor

whenTrue

val whenTrue: TracingTensor

Inheritors

IfThenElse