io.improbable.keanu.vertices.dbl.nonprobabilistic.operators.unary

Class SliceVertex

java.lang.Object

io.improbable.keanu.vertices.Vertex<DoubleTensor>

io.improbable.keanu.vertices.dbl.DoubleVertex

io.improbable.keanu.vertices.dbl.nonprobabilistic.operators.unary.DoubleUnaryOpVertex

io.improbable.keanu.vertices.dbl.nonprobabilistic.operators.unary.SliceVertex

All Implemented Interfaces:

Variable<DoubleTensor,VertexState<DoubleTensor>>, io.improbable.keanu.kotlin.DoubleOperators<DoubleVertex>, io.improbable.keanu.kotlin.NumberOperators<DoubleVertex>, Differentiable, NonProbabilistic<DoubleTensor>, Observable<DoubleTensor>

public class SliceVertex
extends DoubleUnaryOpVertex
implements Differentiable

Field Summary

Fields inherited from class io.improbable.keanu.vertices.dbl.nonprobabilistic.operators.unary.DoubleUnaryOpVertex

INPUT_VERTEX_NAME, inputVertex

Constructor Summary

Constructors

Constructor and Description
SliceVertex(DoubleVertex inputVertex, int dimension, long index) Takes the slice along a given dimension and index of a vertex

Method Summary

Modifier and Type	Method and Description
PartialDerivative	forwardModeAutoDifferentiation(java.util.Map<Vertex,PartialDerivative> derivativeOfParentsWithRespectToInput)
int	getDimension()
long	getIndex()
protected DoubleTensor	op(DoubleTensor value)
java.util.Map<Vertex,PartialDerivative>	reverseModeAutoDifferentiation(PartialDerivative derivativeOfOutputWithRespectToSelf)

Methods inherited from class io.improbable.keanu.vertices.dbl.nonprobabilistic.operators.unary.DoubleUnaryOpVertex

calculate, getInputVertex

Methods inherited from class io.improbable.keanu.vertices.dbl.DoubleVertex

abs, acos, asin, atan, atan2, ceil, concat, cos, div, div, divideBy, divideBy, equalTo, exp, floor, getValue, greaterThan, greaterThanOrEqualTo, lambda, lambda, lessThan, lessThanOrEqualTo, loadValue, log, logGamma, matrixDeterminant, matrixInverse, matrixMultiply, max, min, minus, minus, multiply, multiply, notEqualTo, observe, observe, permute, plus, plus, pow, pow, reshape, reverseDiv, reverseMinus, round, saveValue, setAndCascade, setAndCascade, setValue, setValue, setWithMask, setWithMask, sigmoid, sin, slice, sum, sum, take, tan, times, times, toGreaterThanMask, toGreaterThanMask, toGreaterThanOrEqualToMask, toGreaterThanOrEqualToMask, toInteger, toLessThanMask, toLessThanMask, toLessThanOrEqualToMask, toLessThanOrEqualToMask, transpose, unaryMinus

Methods inherited from class io.improbable.keanu.vertices.Vertex

addChild, addParent, addParents, equals, eval, getChildren, getConnectedGraph, getDegree, getId, getIndentation, getLabel, getObservedValue, getParents, getRank, getReference, getShape, getState, getValue, hashCode, hasValue, isDifferentiable, isObserved, isProbabilistic, lazyEval, observe, observeOwnValue, print, print, removeLabel, save, setAndCascade, setLabel, setLabel, setParents, setParents, setState, setValue, toString, unobserve

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface io.improbable.keanu.vertices.dbl.Differentiable

withRespectToSelf

Methods inherited from interface io.improbable.keanu.vertices.NonProbabilistic

contradictsObservation

Methods inherited from interface io.improbable.keanu.vertices.Observable

getObservedValue, isObserved, observe, unobserve

Constructor Detail

SliceVertex

public SliceVertex(DoubleVertex inputVertex,
                   int dimension,
                   long index)

Takes the slice along a given dimension and index of a vertex

Parameters:

inputVertex - the input vertex

dimension - the dimension to extract along

index - the index of extraction

Method Detail

op

protected DoubleTensor op(DoubleTensor value)

Specified by:

op in class DoubleUnaryOpVertex

reverseModeAutoDifferentiation

public java.util.Map<Vertex,PartialDerivative> reverseModeAutoDifferentiation(PartialDerivative derivativeOfOutputWithRespectToSelf)

Specified by:

reverseModeAutoDifferentiation in interface Differentiable

forwardModeAutoDifferentiation

public PartialDerivative forwardModeAutoDifferentiation(java.util.Map<Vertex,PartialDerivative> derivativeOfParentsWithRespectToInput)

Specified by:

forwardModeAutoDifferentiation in interface Differentiable

getDimension

public int getDimension()

getIndex

public long getIndex()