org.nd4j.linalg.api.ops.impl.shape.tensorops

Class TensorArrayV3

java.lang.Object

org.nd4j.autodiff.functions.DifferentialFunction

org.nd4j.linalg.api.ops.DynamicCustomOp

org.nd4j.linalg.api.ops.impl.shape.tensorops.BaseTensorOp

org.nd4j.linalg.api.ops.impl.shape.tensorops.TensorArrayV3

All Implemented Interfaces:

CustomOp

public class TensorArrayV3
extends BaseTensorOp

Nested Class Summary

Nested classes/interfaces inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp

DynamicCustomOp.DynamicCustomOpsBuilder, DynamicCustomOp.SameDiffBuilder

Field Summary

Fields inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp

iArguments, inplaceCall, inputArguments, outputArguments, outputVariables, tArguments

Fields inherited from class org.nd4j.autodiff.functions.DifferentialFunction

dimensions, extraArgs, inPlace, sameDiff, scalarValue

Constructor Summary

Constructors

Constructor and Description
TensorArrayV3()
TensorArrayV3(SameDiff sameDiff)
TensorArrayV3(String name, SameDiff sameDiff)
TensorArrayV3(TensorArrayV3 ta)
TensorArrayV3(TensorArrayV3 ta, SDVariable[] inputs)

Method Summary

Modifier and Type	Method and Description
SDVariable	concat()
TensorList	execute(SameDiff sameDiff)
SDVariable	gather(int... indices)
SDVariable	gather(SDVariable indices)
SameDiff	getSameDiff()
void	initFromTensorFlow(NodeDef nodeDef, SameDiff initWith, Map<String,AttrValue> attributesForNode, GraphDef graph) Initialize the function from the given NodeDef
String	opName() This method returns op opName as string
Op.Type	opType() The type of the op
SDVariable	read(int index)
SDVariable	read(SDVariable index)
void	scatter(SDVariable value, int... indices)
void	scatter(SDVariable value, SDVariable indices)
SDVariable	stack()
String	tensorflowName() The opName of this function tensorflow
String	toString()
void	unstack(SDVariable value)
void	write(int index, SDVariable value)
void	write(SDVariable index, SDVariable value)

Methods inherited from class org.nd4j.linalg.api.ops.impl.shape.tensorops.BaseTensorOp

doDiff, getArgumentArray, getList, onnxName

Methods inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp

addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, asProperties, assertValidForExecution, builder, calculateOutputShape, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, initFromOnnx, inputArguments, numIArguments, numInputArguments, numOutputArguments, numTArguments, opHash, opNum, outputArguments, outputVariables, outputVariables, populateInputsAndOutputsFromSameDiff, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, sameDiffBuilder, setInputArgument, setOutputArgument, tArgs

Methods inherited from class org.nd4j.autodiff.functions.DifferentialFunction

arg, arg, argNames, args, attributeAdaptersForFunction, configFieldName, diff, dup, equals, f, getNumOutputs, getValue, hashCode, hasPlaceHolderInputs, isConfigProperties, larg, mappingsForFunction, onnxNames, outputVariable, outputVariablesNames, propertiesForFunction, rarg, resolvePropertiesFromSameDiffBeforeExecution, setInstanceId, setValueFor, tensorflowNames

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.nd4j.linalg.api.ops.CustomOp

isInplaceCall

Constructor Detail

TensorArrayV3

public TensorArrayV3(String name,
                     SameDiff sameDiff)

TensorArrayV3

public TensorArrayV3(SameDiff sameDiff)

TensorArrayV3

public TensorArrayV3(TensorArrayV3 ta)

TensorArrayV3

public TensorArrayV3(TensorArrayV3 ta,
                     SDVariable[] inputs)

TensorArrayV3

public TensorArrayV3()

Method Detail

tensorflowName

public String tensorflowName()

Description copied from class: DifferentialFunction

The opName of this function tensorflow

Overrides:

tensorflowName in class DynamicCustomOp

Returns:

initFromTensorFlow

public void initFromTensorFlow(NodeDef nodeDef,
                               SameDiff initWith,
                               Map<String,AttrValue> attributesForNode,
                               GraphDef graph)

Description copied from class: DifferentialFunction

Initialize the function from the given NodeDef

Overrides:

initFromTensorFlow in class BaseTensorOp

execute

public TensorList execute(SameDiff sameDiff)

Specified by:

execute in class BaseTensorOp

toString

public String toString()

Overrides:

toString in class BaseTensorOp

opName

public String opName()

Description copied from class: DynamicCustomOp

This method returns op opName as string

Specified by:

opName in interface CustomOp

Overrides:

opName in class DynamicCustomOp

Returns:

opType

public Op.Type opType()

Description copied from class: DifferentialFunction

The type of the op

Overrides:

opType in class BaseTensorOp

Returns:

getSameDiff

public SameDiff getSameDiff()

read

public SDVariable read(int index)

read

public SDVariable read(SDVariable index)

gather

public SDVariable gather(int... indices)

gather

public SDVariable gather(SDVariable indices)

stack

public SDVariable stack()

concat

public SDVariable concat()

write

public void write(int index,
                  SDVariable value)

write

public void write(SDVariable index,
                  SDVariable value)

scatter

public void scatter(SDVariable value,
                    int... indices)

scatter

public void scatter(SDVariable value,
                    SDVariable indices)

unstack

public void unstack(SDVariable value)