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

Class ConcatenationVertex

java.lang.Object

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

io.improbable.keanu.vertices.dbl.DoubleVertex

io.improbable.keanu.vertices.dbl.nonprobabilistic.operators.multiple.ConcatenationVertex

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 ConcatenationVertex
extends DoubleVertex
implements Differentiable, NonProbabilistic<DoubleTensor>

Constructor Summary

Constructors

Constructor and Description
ConcatenationVertex(int dimension, DoubleVertex... operands) A vertex that can concatenate any amount of vertices along a given dimension.
ConcatenationVertex(int dimension, Vertex[] operands)

Method Summary

Modifier and Type	Method and Description
DoubleTensor	calculate()
static PartialDerivative	concat(java.util.List<PartialDerivative> partialsOfOperands, java.util.List<DoubleTensor> operandValues, int dimension)
PartialDerivative	forwardModeAutoDifferentiation(java.util.Map<Vertex,PartialDerivative> derivativeOfParentsWithRespectToInput)
int	getDimension()
DoubleVertex[]	getOperands()
protected DoubleTensor	op(DoubleTensor... inputs)
java.util.Map<Vertex,PartialDerivative>	reverseModeAutoDifferentiation(PartialDerivative derivativeOfOutputWithRespectToSelf)

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

ConcatenationVertex

public ConcatenationVertex(int dimension,
                           DoubleVertex... operands)

A vertex that can concatenate any amount of vertices along a given dimension.

Parameters:

dimension - the dimension to concatenate on. This is the only dimension in which sizes may be different. Negative dimension indexing is not supported.

operands - the operands vertices to concatenate

ConcatenationVertex

public ConcatenationVertex(int dimension,
                           Vertex[] operands)

Method Detail

forwardModeAutoDifferentiation

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

Specified by:

forwardModeAutoDifferentiation in interface Differentiable

concat

public static PartialDerivative concat(java.util.List<PartialDerivative> partialsOfOperands,
                                       java.util.List<DoubleTensor> operandValues,
                                       int dimension)

reverseModeAutoDifferentiation

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

Specified by:

reverseModeAutoDifferentiation in interface Differentiable

calculate

public DoubleTensor calculate()

Specified by:

calculate in interface NonProbabilistic<DoubleTensor>

op

protected DoubleTensor op(DoubleTensor... inputs)

getOperands

public DoubleVertex[] getOperands()

getDimension

public int getDimension()