org.nd4j.linalg.api.ops

Class BaseModule

java.lang.Object

org.nd4j.autodiff.functions.DifferentialFunction

org.nd4j.linalg.api.ops.DynamicCustomOp

org.nd4j.linalg.api.ops.BaseModule

All Implemented Interfaces:

CustomOp, Module

Direct Known Subclasses:

Linear

public abstract class BaseModule
extends DynamicCustomOp
implements Module

Abstract base class for Module that handles Dynamic ops and handles nesting. This is a logical unit for defining layers very similar to pytorch's modules, or tensorflow's layers.

Author:

Adam Gibson

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
BaseModule(String opName, INDArray[] inputs, INDArray[] outputs, List<Double> tArguments, List<Integer> iArguments, List<Module> modules)
BaseModule(String opName, SameDiff sameDiff, SDVariable[] args, boolean inPlace, List<Module> modules)

Method Summary

Modifier and Type	Method and Description
void	addModule(Module module)
Module[]	subModules()

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

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

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.Module

exec, execSameDiff

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

addIArgument, addIArgument, addInputArgument, addOutputArgument, addTArgument, assertValidForExecution, calculateOutputShape, getDescriptor, getIArgument, getInputArgument, getOutputArgument, getTArgument, iArgs, inputArguments, isInplaceCall, numIArguments, numInputArguments, numOutputArguments, numTArguments, opHash, opName, outputArguments, populateInputsAndOutputsFromSameDiff, removeIArgument, removeInputArgument, removeOutputArgument, removeTArgument, tArgs

Constructor Detail

BaseModule

public BaseModule(String opName,
                  INDArray[] inputs,
                  INDArray[] outputs,
                  List<Double> tArguments,
                  List<Integer> iArguments,
                  List<Module> modules)

BaseModule

public BaseModule(String opName,
                  SameDiff sameDiff,
                  SDVariable[] args,
                  boolean inPlace,
                  List<Module> modules)

Method Detail

subModules

public Module[] subModules()

Specified by:

subModules in interface Module

addModule

public void addModule(Module module)

Specified by:

addModule in interface Module