org.nd4j.linalg.api.ops

Interface Module

All Superinterfaces:

CustomOp

All Known Implementing Classes:

BaseModule, Linear

public interface Module
extends CustomOp

A Module is a CustomOp with varying input arguments and automatically calculated outputs defined at a higher level than c++. A Module is meant to be a way of implementing custom operations in straight java/nd4j.

Method Summary

Modifier and Type	Method and Description
void	addModule(Module module)
void	exec(INDArray... inputs)
void	execSameDiff(SDVariable... input)
Module[]	subModules()

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

Method Detail

exec

void exec(INDArray... inputs)

Parameters:

inputs -

subModules

Module[] subModules()

addModule

void addModule(Module module)

execSameDiff

void execSameDiff(SDVariable... input)