api/org.diffkt.tracing/TracingTensor

TracingTensor

interface TracingTensor : DTensor, Traceable

Types

Atan

open class Atan(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

AvgPool

open class AvgPool(x: TracingTensor, poolHeight: Int, poolWidth: Int) : TracingTensor

AvgPoolGrad

open class AvgPoolGrad(x: TracingTensor, poolHeight: Int, poolWidth: Int) : TracingTensor

BroadcastTo

class BroadcastTo(x: TracingTensor, shape: Shape) : TracingTensor.Companion.TracingTensorBase

Companion

object Companion

Compare

open class Compare(left: TracingTensor, right: TracingTensor, comparison: ComparisonKind) : TracingTensor.Companion.TracingTensorBase

Concat

class Concat(slices: List<TracingTensor>, axis: Int) : TracingTensor.Companion.TracingTensorBase

Constant

open class Constant(values: DTensor) : TracingTensor.Companion.TracingTensorBase

ConvImpl

class ConvImpl(signal: TracingTensor, filter: TracingTensor, hStride: Int, vStride: Int, padding: Convolve.Padding2D) : TracingTensor

Cos

open class Cos(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Digamma

open class Digamma(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Div

open class Div(left: TracingTensor, right: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Exp

open class Exp(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Expand

class Expand(x: TracingTensor, shape: Shape) : TracingTensor.Companion.TracingTensorBase

Flip

class Flip(x: TracingTensor, axes: IntArray) : TracingTensor.Companion.TracingTensorBase

Gather

class Gather(x: TracingTensor, indexes: List<Int>, axis: Int, paddingIndex: Int) : TracingTensor.Companion.TracingTensorBase

GatherAtIndices

class GatherAtIndices(x: TracingTensor, indexes: List<IntArray>) : TracingTensor.Companion.TracingTensorBase

IdentityGradient

class IdentityGradient(halfShape: Shape) : TracingTensor.Companion.TracingTensorBase

IfThenElse

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

Lgamma

open class Lgamma(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Ln

open class Ln(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

LogSoftmax

class LogSoftmax(x: TracingTensor, axis: Int) : TracingTensor.Companion.TracingTensorBase

LogSoftmaxGrad

class LogSoftmaxGrad(x: TracingTensor, axis: Int, logSoftmax: TracingTensor, upstream: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Matmul

open class Matmul(x: TracingTensor, y: TracingTensor, a: Shape, b: Shape, c: Shape, d: Shape) : TracingTensor.Companion.TracingTensorBase

MaxPoolWithIndices

class MaxPoolWithIndices(x: TracingTensor, poolHeight: Int, poolWidth: Int, withIndices: Boolean) : TracingTensor

Meld

class Meld(values: List<TracingTensor>) : TracingTensor.Companion.TracingTensorBase

Minus

open class Minus(left: TracingTensor, right: TracingTensor) : TracingTensor.Companion.TracingTensorBase

OuterProduct

open class OuterProduct(x: TracingTensor, y: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Plus

open class Plus(left: TracingTensor, right: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Polygamma

open class Polygamma(n: Int, x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Pow

open class Pow(base: TracingTensor, exponent: Float) : TracingTensor.Companion.TracingTensorBase

RandomFloats

open class RandomFloats(key: TracingRandomKey, shape: Shape) : TracingTensor.Companion.TracingTensorBase

Relu

open class Relu(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

ReluGrad

open class ReluGrad(x: TracingTensor, upstream: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Reshape

class Reshape(x: TracingTensor, shape: Shape) : TracingTensor.Companion.TracingTensorBase

Scatter

class Scatter(x: TracingTensor, indexes: List<Int>, axis: Int, newShape: Shape, paddingIndex: Int) : TracingTensor.Companion.TracingTensorBase

ScatterAtIndices

class ScatterAtIndices(x: TracingTensor, indexes: List<IntArray>, shape: Shape) : TracingTensor.Companion.TracingTensorBase

Sigmoid

open class Sigmoid(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Sin

open class Sin(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Split

open class Split(x: TracingTensor, shapes: List<Shape>) : TracingTensor.Companion.TracingTensorBase

SplitPart

open class SplitPart(from: TracingTensor, index: Int, shape: Shape) : TracingTensor.Companion.TracingTensorBase

Sqrt

open class Sqrt(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Squeeze

open class Squeeze(x: TracingTensor, axis: Int) : TracingTensor.Companion.TracingTensorBase

Sum

open class Sum(x: TracingTensor, axes: IntArray, keepDims: Boolean) : TracingTensor.Companion.TracingTensorBase

Tan

open class Tan(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Tanh

open class Tanh(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Times

open class Times(left: TracingTensor, right: TracingTensor) : TracingTensor.Companion.TracingTensorBase

TimesScalar

open class TimesScalar(left: TracingScalar, right: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Transpose

class Transpose(x: TracingTensor, axes: IntArray) : TracingTensor.Companion.TracingTensorBase

UnaryMinus

open class UnaryMinus(x: TracingTensor) : TracingTensor.Companion.TracingTensorBase

Unsqueeze

class Unsqueeze(x: TracingTensor, axis: Int) : TracingTensor.Companion.TracingTensorBase

Variable

open class Variable(varIndex: Int, name: String?, shape: Shape, traceId: TraceId) : TracingTensor.Companion.TracingTensorBase

View1

open class View1(x: TracingTensor, indexes: IntArray, shape: Shape) : TracingTensor.Companion.TracingTensorBase

View2

open class View2(x: TracingTensor, index: Int, axis: Int, shape: Shape) : TracingTensor.Companion.TracingTensorBase

View3

class View3(x: TracingTensor, index: IntRange, axis: Int) : TracingTensor.Companion.TracingTensorBase

Zero

open class Zero(shape: Shape) : TracingTensor.Companion.TracingTensorBase

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

PrintedTensor

TracingScalar

TracingTensorBase

ConvImpl

AvgPool

AvgPoolGrad

MaxPoolWithIndices