api/org.diffkt.tracing/TracingTensorOperations

TracingTensorOperations

object TracingTensorOperations : Operations

Functions

atan

open override fun atan(x: DTensor): DTensor

avgPool

open override fun avgPool(x: DTensor, poolHeight: Int, poolWidth: Int): DTensor

avgPoolGrad

open override fun avgPoolGrad(x: DTensor, poolHeight: Int, poolWidth: Int): DTensor

batchNorm

open fun batchNorm(input: DTensor, scaleShift: DTensor, derivativeId: DerivativeID): BatchNormResult

broadcastTo

open override fun broadcastTo(x: DTensor, newShape: Shape): DTensor

compare

open override fun compare(left: DTensor, right: DTensor, comparison: ComparisonKind): DTensor

concat

open override fun concat(slices: List<DTensor>, axis: Int, derivativeId: DerivativeID): DTensor

open override fun concat(left: DTensor, right: DTensor, axis: Int, derivativeId: DerivativeID): DTensor

convImpl

open override fun convImpl(signal: DTensor, filter: DTensor, hStride: Int, vStride: Int, padding: Convolve.Padding2D, derivativeId: DerivativeID): DTensor

Applies convolution to the tensor signal, using filter. Both signal and filter must be of rank 4. The expected shape of signal is NHWC (num signal, height, width, channels) and the expected filter shape is OHWI (output channels, height, width, input channels) where C == I

cos

open override fun cos(x: DTensor): DTensor

digamma

open override fun digamma(x: DTensor): DTensor

div

open override fun div(left: DTensor, right: DTensor, derivativeId: DerivativeID): DTensor

exp

open override fun exp(x: DTensor): DTensor

expand

open override fun expand(x: DTensor, newShape: Shape): DTensor

flip

open override fun flip(x: DTensor, axes: IntArray): DTensor

gamma

open override fun gamma(alpha: DTensor, randomKey: RandomKey): DTensor

gather

open override fun gather(x: DTensor, indices: List<Int>, axis: Int, paddingIndex: Int): DTensor

gatherAtIndices

open override fun gatherAtIndices(x: DTensor, indices: List<IntArray>): DTensor

identityGradientOfSameKind

open override fun identityGradientOfSameKind(x: DTensor, halfShape: Shape): DTensor

ifThenElse

open override fun ifThenElse(condition: DTensor, whenTrue: DTensor, whenFalse: DTensor, derivativeId: DerivativeID): DTensor

lgamma

open override fun lgamma(x: DTensor): DTensor

open override fun ln(x: DTensor): DTensor

logSoftmax

open override fun logSoftmax(x: DTensor, axis: Int): DTensor

logSoftmaxGrad

open override fun logSoftmaxGrad(x: DTensor, axis: Int, logSoftmax: DTensor, upstream: DTensor): DTensor

matmul

open override fun matmul(x: DTensor, y: DTensor, a: Shape, b: Shape, c: Shape, d: Shape, derivativeId: DerivativeID): DTensor

maxPoolWithIndices

open override fun maxPoolWithIndices(x: DTensor, poolHeight: Int, poolWidth: Int, withIndices: Boolean): Pair<DTensor, List<IntArray>?>

meld

open override fun meld(values: List<DTensor>, derivativeId: DerivativeID): DTensor

minus

open override fun minus(left: DTensor, right: DTensor, derivativeId: DerivativeID): DTensor

outerProduct

open override fun outerProduct(x: DTensor, y: DTensor, derivativeId: DerivativeID): DTensor

plus

open override fun plus(left: DTensor, right: DTensor, derivativeId: DerivativeID): DTensor

polygamma

open override fun polygamma(n: Int, x: DTensor): DTensor

pow

open override fun pow(base: DTensor, exponent: Float): DTensor

relu

open override fun relu(x: DTensor): DTensor

reluGrad

open override fun reluGrad(x: DTensor, reluUpstream: DTensor, derivativeId: DerivativeID): DTensor

reshape

open override fun reshape(x: DTensor, newShape: Shape): DTensor

reshapeToScalar

open override fun reshapeToScalar(x: DTensor): DScalar

scatter

open override fun scatter(x: DTensor, indices: List<Int>, axis: Int, newShape: Shape, paddingIndex: Int): DTensor

scatterAtIndices

open override fun scatterAtIndices(x: DTensor, indices: List<IntArray>, newShape: Shape): DTensor

sigmoid

open override fun sigmoid(x: DTensor): DTensor

sin

open override fun sin(x: DTensor): DTensor

split

open override fun split(x: DTensor, shapes: List<Shape>): List<DTensor>

sqrt

open override fun sqrt(x: DTensor): DTensor

squeeze

open override fun squeeze(x: DTensor, axis: Int): DTensor

sum

open override fun sum(x: DTensor, axes: IntArray, keepDims: Boolean): DTensor

tan

open override fun tan(x: DTensor): DTensor

tanh

open override fun tanh(x: DTensor): DTensor

times

open override fun times(left: DTensor, right: DTensor, derivativeId: DerivativeID): DTensor

timesScalar

open override fun timesScalar(left: DScalar, right: DTensor, derivativeId: DerivativeID): DTensor

transpose

open override fun transpose(x: DTensor, axes: IntArray): DTensor

unaryMinus

open override fun unaryMinus(x: DTensor): DTensor

unsqueeze

open override fun unsqueeze(x: DTensor, axis: Int): DTensor

open override fun view1(x: DTensor, indices: IntArray): DTensor

open override fun view2(x: DTensor, index: Int, axis: Int): DTensor

open override fun view3(x: DTensor, index: IntRange, axis: Int): DTensor

wrap

fun wrap(x: DTensor): TracingTensor

zeroOfSameKind

open override fun zeroOfSameKind(x: DTensor, shape: Shape): DTensor

Properties

name

open override val name: String

tensor

open val tensor: Operations

For a scalar operations, returns the corresponding tensor operations.