Package | Description |
---|---|
org.nd4j.linalg.api.ops |
Modifier and Type | Class and Description |
---|---|
static class |
DynamicCustomOp.SameDiffBuilder |
Modifier and Type | Method and Description |
---|---|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addFloatingPointArguments(Double... targs)
This method takes arbitrary number of Double arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.SameDiffBuilder.addInputs(DifferentialFunction... inputs) |
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.SameDiffBuilder.addInputs(INDArray... inputs) |
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addInputs(INDArray... inputs)
This method
takes arbitrary number of input INDArrays in, as Op input
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addIntegerArguments(int... iargs)
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addIntegerArguments(int arg)
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addIntegerArguments(List<Integer> iargs)
This method takes arbitrary number of Integer arguments for op,
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.SameDiffBuilder.addOutputs(DifferentialFunction... outputs) |
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.SameDiffBuilder.addOutputs(INDArray... outputs) |
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addOutputs(INDArray... outputs)
This method takes arbitrary number of
output INDArrays in, to store operation result
Note that this ACCUMULATES arguments.
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.addOutputShape(int[] shape)
Adds an oup
|
static DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.builder(String opName)
This method takes custom opname, and return Op DynamicCustomOpsBuilder instance
|
DynamicCustomOp.DynamicCustomOpsBuilder |
DynamicCustomOp.DynamicCustomOpsBuilder.callInplace(boolean reallyCall)
Whether an op call is in place or not.
|
Copyright © 2018. All rights reserved.