api/org.diffkt.tracing/TracingScalar

TracingScalar

interface TracingScalar : TracingTensor, DScalar

Types

Atan

class Atan(x: TracingScalar) : TracingTensor.Atan, TracingScalar

Compare

class Compare(left: TracingTensor, right: TracingTensor, comparison: ComparisonKind) : TracingTensor.Compare, TracingScalar

Constant

class Constant(values: DScalar) : TracingTensor.Constant, TracingScalar

Cos

class Cos(x: TracingScalar) : TracingTensor.Cos, TracingScalar

Digamma

class Digamma(x: TracingScalar) : TracingTensor.Digamma, TracingScalar

Div

class Div(left: TracingScalar, right: TracingScalar) : TracingTensor.Div, TracingScalar

Exp

class Exp(x: TracingScalar) : TracingTensor.Exp, TracingScalar

FloatScalar

class FloatScalar(key: TracingRandomKey) : TracingTensor.RandomFloats, TracingScalar

IfThenElse

class IfThenElse(cond: TracingTensor, whenTrue: TracingTensor, whenFalse: TracingTensor) : TracingTensor.IfThenElse, TracingScalar

Lgamma

class Lgamma(x: TracingScalar) : TracingTensor.Lgamma, TracingScalar

Ln

class Ln(x: TracingScalar) : TracingTensor.Ln, TracingScalar

Matmul

class Matmul(x: TracingTensor, y: TracingTensor, a: Shape, b: Shape, c: Shape, d: Shape) : TracingTensor.Matmul, TracingScalar

Minus

class Minus(left: TracingScalar, right: TracingScalar) : TracingTensor.Minus, TracingScalar

Plus

class Plus(left: TracingScalar, right: TracingScalar) : TracingTensor.Plus, TracingScalar

Polygamma

class Polygamma(n: Int, x: TracingScalar) : TracingTensor.Polygamma, TracingScalar

Pow

class Pow(base: TracingScalar, exponent: Float) : TracingTensor.Pow, TracingScalar

Relu

class Relu(x: TracingTensor) : TracingTensor.Relu, TracingScalar

ReluGrad

class ReluGrad(x: TracingTensor, upstream: TracingTensor) : TracingTensor.ReluGrad, TracingScalar

ReshapeToScalar

class ReshapeToScalar(x: TracingTensor) : TracingScalar

Sigmoid

class Sigmoid(x: TracingTensor) : TracingTensor.Sigmoid, TracingScalar

Sin

class Sin(x: TracingScalar) : TracingTensor.Sin, TracingScalar

Split

class Split(from: TracingScalar, shapes: List<Shape>) : TracingTensor.Split, TracingScalar

SplitPart

class SplitPart(from: TracingTensor, index: Int) : TracingTensor.SplitPart, TracingScalar

Sqrt

class Sqrt(x: TracingScalar) : TracingTensor.Sqrt, TracingScalar

Squeeze

class Squeeze(x: TracingTensor) : TracingTensor.Squeeze, TracingScalar

Sum

open class Sum(x: TracingTensor) : TracingTensor.Sum, TracingScalar

Tan

class Tan(x: TracingScalar) : TracingTensor.Tan, TracingScalar

Tanh

class Tanh(x: TracingScalar) : TracingTensor.Tanh, TracingScalar

TimesScalar

class TimesScalar(left: TracingScalar, right: TracingScalar) : TracingTensor.TimesScalar, TracingScalar

UnaryMinus

class UnaryMinus(x: TracingScalar) : TracingTensor.UnaryMinus, TracingScalar

Variable

class Variable(varIndex: Int, name: String?, traceId: TraceId) : TracingTensor.Variable, TracingScalar

View1

class View1(x: TracingTensor, indexes: IntArray) : TracingTensor.View1, TracingScalar

View2

class View2(x: TracingTensor, index: Int, axis: Int) : TracingTensor.View2, TracingScalar

Zero

class Zero : TracingTensor.Zero, TracingScalar

Functions

accept

abstract fun <R> accept(v: TracingVisitor<R>): R

floatEval

open fun floatEval(vars: FloatArray): Float

get

open operator fun get(index: Int): DTensor

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

toCodeString

open fun toCodeString(): String

wrap

open override fun wrap(wrapper: Wrapper): DTensor

Wrapper around the tensor

Properties

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.

primal

open override val primal: DScalar

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 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.

Inheritors

Constant

Variable

Plus

Minus

TimesScalar

Div

Zero

UnaryMinus

Matmul

Sin

Cos

Tan

Atan

Exp

Ln

Lgamma

Digamma

Polygamma

Sqrt

Tanh

Split

SplitPart

Pow

View1

View2

ReshapeToScalar

Squeeze

Relu

ReluGrad

Sigmoid

Sum

Compare

IfThenElse

FloatScalar