Operations

interface Operations

Functions

atan

abstract fun atan(x: DTensor): DTensor

avgPool

abstract fun avgPool(x: DTensor, poolHeight: Int, poolWidth: Int): DTensor

avgPoolGrad

abstract fun avgPoolGrad(x: DTensor, poolHeight: Int, poolWidth: Int): DTensor

batchNorm

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

broadcastTo

abstract fun broadcastTo(x: DTensor, newShape: Shape): DTensor

compare

abstract fun compare(left: DTensor, right: DTensor, comparison: ComparisonKind): DTensor

concat

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

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

convImpl

abstract 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

abstract fun cos(x: DTensor): DTensor

digamma

abstract fun digamma(x: DTensor): DTensor

div

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

exp

abstract fun exp(x: DTensor): DTensor

expand

abstract fun expand(x: DTensor, newShape: Shape): DTensor

flip

abstract fun flip(x: DTensor, axes: IntArray): DTensor

gamma

abstract fun gamma(alpha: DTensor, randomKey: RandomKey): DTensor

gather

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

gatherAtIndices

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

identityGradientOfSameKind

abstract fun identityGradientOfSameKind(x: DTensor, halfShape: Shape): DTensor

ifThenElse

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

lgamma

abstract fun lgamma(x: DTensor): DTensor

abstract fun ln(x: DTensor): DTensor

logSoftmax

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

logSoftmaxGrad

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

matmul

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

maxPoolWithIndices

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

meld

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

minus

abstract fun minus(left: DTensor, right: DTensor, derivativeId: DerivativeID): DTensor

outerProduct

abstract fun outerProduct(x: DTensor, y: DTensor, derivativeId: DerivativeID): DTensor

plus

abstract fun plus(left: DTensor, right: DTensor, derivativeId: DerivativeID): DTensor

polygamma

abstract fun polygamma(n: Int, x: DTensor): DTensor

pow

abstract fun pow(base: DTensor, exponent: Float): DTensor

relu

abstract fun relu(x: DTensor): DTensor

reluGrad

abstract fun reluGrad(x: DTensor, reluUpstream: DTensor, derivativeId: DerivativeID): DTensor

reshape

abstract fun reshape(x: DTensor, newShape: Shape): DTensor

reshapeToScalar

abstract fun reshapeToScalar(x: DTensor): DScalar

scatter

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

scatterAtIndices

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

sigmoid

abstract fun sigmoid(x: DTensor): DTensor

sin

abstract fun sin(x: DTensor): DTensor

split

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

sqrt

abstract fun sqrt(x: DTensor): DTensor

squeeze

abstract fun squeeze(x: DTensor, axis: Int): DTensor

sum

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

tan

abstract fun tan(x: DTensor): DTensor

tanh

abstract fun tanh(x: DTensor): DTensor

times

abstract fun times(left: DTensor, right: DTensor, derivativeId: DerivativeID): DTensor

timesScalar

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

transpose

abstract fun transpose(x: DTensor, axes: IntArray): DTensor

unaryMinus

abstract fun unaryMinus(x: DTensor): DTensor

unsqueeze

abstract fun unsqueeze(x: DTensor, axis: Int): DTensor

abstract fun view1(x: DTensor, indices: IntArray): DTensor

abstract fun view2(x: DTensor, index: Int, axis: Int): DTensor

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

zeroOfSameKind

abstract fun zeroOfSameKind(x: DTensor, shape: Shape): DTensor

Properties

name

abstract val name: String

tensor

open val tensor: Operations

For a scalar operations, returns the corresponding tensor operations.

Inheritors

TracingTensorOperations