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<OUT>

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

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

All Implemented Interfaces:

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

public class GenericSliceVertex<T>
extends UnaryOpVertex<Tensor<T>,Tensor<T>>

Field Summary

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

INPUT_NAME, inputVertex

Constructor Summary

Constructors

Constructor and Description
GenericSliceVertex(Vertex<Tensor<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 Tensor<T>	op(Tensor<T> input)

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

calculate, getInputVertex

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<Tensor<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 Tensor<T> op(Tensor<T> input)

Specified by:

op in class UnaryOpVertex<Tensor<T>,Tensor<T>>

getDimension

public int getDimension()

getIndex

public int getIndex()