| 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[] |
Module.subModules() |
Module[] |
BaseModule.subModules() |
| Modifier and Type | Method and Description |
|---|---|
void |
Module.addModule(Module module) |
void |
BaseModule.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.