org.nd4j.linalg.api.ops

Class DynamicCustomOp.SameDiffBuilder

java.lang.Object

org.nd4j.linalg.api.ops.DynamicCustomOp.DynamicCustomOpsBuilder

org.nd4j.linalg.api.ops.DynamicCustomOp.SameDiffBuilder

Enclosing class:

DynamicCustomOp

public static class DynamicCustomOp.SameDiffBuilder
extends DynamicCustomOp.DynamicCustomOpsBuilder

Field Summary

Fields inherited from class org.nd4j.linalg.api.ops.DynamicCustomOp.DynamicCustomOpsBuilder

inplaceAllowed, inplaceCall, numIArguments, numInputs, numOutputs, numTArguments, opHash, opName, outputShapes

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	SameDiffBuilder(String opName, SameDiff sameDiff, long hash, int numInputs, int numOutputs, boolean inplaceAllowed, int numTArguments, int numIArguments)

Method Summary

Modifier and Type	Method and Description
DynamicCustomOp.DynamicCustomOpsBuilder	addInputs(DifferentialFunction... inputs)
DynamicCustomOp.DynamicCustomOpsBuilder	addInputs(INDArray... inputs) This method takes arbitrary number of input INDArrays in, as Op input Note that this ACCUMULATES arguments.
DynamicCustomOp.DynamicCustomOpsBuilder	addOutputs(DifferentialFunction... outputs)
DynamicCustomOp.DynamicCustomOpsBuilder	addOutputs(INDArray... outputs) This method takes arbitrary number of output INDArrays in, to store operation result Note that this ACCUMULATES arguments.
DynamicCustomOp	build()
DynamicCustomOp.SameDiffBuilder	sameDiff(SameDiff sameDiff)

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

addFloatingPointArguments, addIntegerArguments, addIntegerArguments, addIntegerArguments, addOutputShape, callInplace

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SameDiffBuilder

protected SameDiffBuilder(String opName,
                          SameDiff sameDiff,
                          long hash,
                          int numInputs,
                          int numOutputs,
                          boolean inplaceAllowed,
                          int numTArguments,
                          int numIArguments)

Method Detail

sameDiff

public DynamicCustomOp.SameDiffBuilder sameDiff(SameDiff sameDiff)

addInputs

public DynamicCustomOp.DynamicCustomOpsBuilder addInputs(INDArray... inputs)

Description copied from class: DynamicCustomOp.DynamicCustomOpsBuilder

This method takes arbitrary number of input INDArrays in, as Op input Note that this ACCUMULATES arguments. You are able to call this method multiple times and it will add arguments to a list. PLEASE NOTE: this method does NOT validate lengths/shapes.

Overrides:

addInputs in class DynamicCustomOp.DynamicCustomOpsBuilder

Returns:

addOutputs

public DynamicCustomOp.DynamicCustomOpsBuilder addOutputs(INDArray... outputs)

Description copied from class: DynamicCustomOp.DynamicCustomOpsBuilder

This method takes arbitrary number of output INDArrays in, to store operation result Note that this ACCUMULATES arguments. You are able to call this method multiple times and it will add arguments to a list. PLEASE NOTE: this method does NOT validate lengths/shapes.

Overrides:

addOutputs in class DynamicCustomOp.DynamicCustomOpsBuilder

Returns:

addInputs

public DynamicCustomOp.DynamicCustomOpsBuilder addInputs(DifferentialFunction... inputs)

addOutputs

public DynamicCustomOp.DynamicCustomOpsBuilder addOutputs(DifferentialFunction... outputs)

build

public DynamicCustomOp build()

Overrides:

build in class DynamicCustomOp.DynamicCustomOpsBuilder