Package | Description |
---|---|
org.nd4j.linalg.api.ops | |
org.nd4j.linalg.api.ops.impl.layers |
Modifier and Type | Class and Description |
---|---|
class |
BaseModule
Abstract base class for
Module
that handles Dynamic ops and handles nesting. |
Modifier and Type | Method and Description |
---|---|
Module[] |
BaseModule.subModules() |
Module[] |
Module.subModules() |
Modifier and Type | Method and Description |
---|---|
void |
BaseModule.addModule(Module module) |
void |
Module.addModule(Module module) |
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) |
Modifier and Type | Class and Description |
---|---|
class |
Linear
Linear:
a * bT
|
Copyright © 2018. All rights reserved.