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

Class GenericTakeVertex<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.GenericTakeVertex<T>

All Implemented Interfaces:

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

public class GenericTakeVertex<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
GenericTakeVertex(Vertex<GenericTensor<T>> inputVertex, long... index) A vertex that extracts a scalar at a given index

Method Summary

Modifier and Type	Method and Description
long[]	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

GenericTakeVertex

public GenericTakeVertex(Vertex<GenericTensor<T>> inputVertex,
                         long... index)

A vertex that extracts a scalar at a given index

Parameters:

inputVertex - the input vertex

index - the index of extraction

Method Detail

op

protected GenericTensor<T> op(GenericTensor<T> input)

Specified by:

op in class GenericTensorUnaryOpVertex<T,T>

getIndex

public long[] getIndex()