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

Class GenericSliceVertex<T>

java.lang.Object

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

io.improbable.keanu.vertices.generic.GenericVertex<GenericTensor<T>>

io.improbable.keanu.vertices.generic.GenericTensorVertex<OUT>

io.improbable.keanu.vertices.generic.nonprobabilistic.operators.unary.GenericTensorUnaryOpVertex<T,T>

io.improbable.keanu.vertices.generic.nonprobabilistic.operators.unary.GenericSliceVertex<T>

All Implemented Interfaces:

Variable<GenericTensor<T>,VertexState<GenericTensor<T>>>, NonProbabilistic<GenericTensor<T>>, Observable<GenericTensor<T>>

public class GenericSliceVertex<T>
extends GenericTensorUnaryOpVertex<T,T>

Field Summary

Fields inherited from class io.improbable.keanu.vertices.generic.nonprobabilistic.operators.unary.GenericTensorUnaryOpVertex

INPUT_NAME, inputVertex

Constructor Summary

Constructors

Constructor and Description
GenericSliceVertex(Vertex<GenericTensor<T>> inputVertex, int dimension, int index) Takes the slice along a given dimension and index of a vertex

Method Summary

Modifier and Type	Method and Description
int	getDimension()
int	getIndex()
protected GenericTensor<T>	op(GenericTensor<T> input)

Methods inherited from class io.improbable.keanu.vertices.generic.nonprobabilistic.operators.unary.GenericTensorUnaryOpVertex

calculate, getInputVertex

Methods inherited from class io.improbable.keanu.vertices.generic.GenericTensorVertex

slice, take

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, loadValue, observe, observeOwnValue, print, print, removeLabel, save, saveValue, 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.NonProbabilistic

contradictsObservation

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

getObservedValue, isObserved, observe, unobserve

Constructor Detail

GenericSliceVertex

public GenericSliceVertex(Vertex<GenericTensor<T>> inputVertex,
                          int dimension,
                          int 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 GenericTensor<T> op(GenericTensor<T> input)

Specified by:

op in class GenericTensorUnaryOpVertex<T,T>

getDimension

public int getDimension()

getIndex

public int getIndex()