org.nd4j.linalg.api.ops.factory

Class DefaultOpFactory

java.lang.Object

org.nd4j.linalg.api.ops.factory.DefaultOpFactory

All Implemented Interfaces:

OpFactory

public class DefaultOpFactory
extends Object
implements OpFactory

Default operations factory

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
DefaultOpFactory()

Method Summary

Modifier and Type	Method and Description
Accumulation	createAccum(String name, INDArray x) Accumulation operation
Accumulation	createAccum(String name, INDArray x, INDArray y)
Accumulation	createAccum(String name, INDArray x, INDArray y, INDArray z) Accumulation operation
Accumulation	createAccum(String name, INDArray x, INDArray y, INDArray z, Object[] extraArgs)
BroadcastOp	createBroadcastOp(String name, INDArray x, INDArray y, INDArray z, int... dimension) Create a vector operation
BroadcastOp	createBroadcastOp(String name, INDArray x, INDArray y, INDArray z, Object[] extraArgs, int... dimension)
BroadcastOp	createBroadcastOp(String name, INDArray x, INDArray y, int... dimension) Create a vector operation
GradientOp	createGradientOp(String name, INDArray x, INDArray y, INDArray z)
IndexAccumulation	createIndexAccum(String name, INDArray x)
IndexAccumulation	createIndexAccum(String name, INDArray x, INDArray y) Index accumulation operation
IndexAccumulation	createIndexAccum(String opName, INDArray x, INDArray y, INDArray z, Object[] extraArgs)
LossFunction	createLossFunction(String name, INDArray x, INDArray y) Create a loss function with the given inputs and outputs
ScalarOp	createScalarTransform(String name, INDArray x, double scalar)
ScalarOp	createScalarTransform(String name, INDArray x, INDArray y, double scalar)
ScalarOp	createScalarTransform(String name, INDArray x, INDArray y, INDArray z, double scalar)
ScalarOp	createScalarTransform(String name, INDArray x, INDArray y, INDArray z, Object[] extraArgs, double scalar)
ScalarOp	createScalarTransform(String name, INDArray x, Object[] extraArgs, double scalar)
Op	createShape(String name, INDArray x, INDArray z, Object[] extraArgs)
TransformOp	createTransform(String name, INDArray x)
TransformOp	createTransform(String name, INDArray x, INDArray y)
TransformOp	createTransform(String name, INDArray x, INDArray y, INDArray z)
TransformOp	createTransform(String name, INDArray x, INDArray y, INDArray z, Object[] extraArgs)
TransformOp	createTransform(String name, INDArray x, Object[] extraArgs)
Op	getOpByName(String opName) This method returns Op instance if opName exists, null otherwise
int	getOpNumByName(String opName) This method returns op id number for given opName
int	getOpNumIfExists(String opName) This method returns op id number if opName exists, or -1 otherwise

Methods inherited from class java.lang.Object

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

Constructor Detail

DefaultOpFactory

public DefaultOpFactory()

Method Detail

createGradientOp

public GradientOp createGradientOp(String name,
                                   INDArray x,
                                   INDArray y,
                                   INDArray z)

Specified by:

createGradientOp in interface OpFactory

createShape

public Op createShape(String name,
                      INDArray x,
                      INDArray z,
                      Object[] extraArgs)

Specified by:

createShape in interface OpFactory

Parameters:

name -

x -

z -

Returns:

createLossFunction

public LossFunction createLossFunction(String name,
                                       INDArray x,
                                       INDArray y)

Description copied from interface: OpFactory

Create a loss function with the given inputs and outputs

Specified by:

createLossFunction in interface OpFactory

Parameters:

name - the opName of the function

x - the input

y - the output

Returns:

a loss function representing the delta between the 2

createAccum

public Accumulation createAccum(String name,
                                INDArray x)

Description copied from interface: OpFactory

Accumulation operation

Specified by:

createAccum in interface OpFactory

Parameters:

name - the opName of the function to create

x - the input to the function

Returns:

the operation

createAccum

public Accumulation createAccum(String name,
                                INDArray x,
                                INDArray y,
                                INDArray z)

Description copied from interface: OpFactory

Accumulation operation

Specified by:

createAccum in interface OpFactory

Parameters:

name - the opName of the function

x - the input

y - the pairwise transformation

z - the output

Returns:

the operation

createAccum

public Accumulation createAccum(String name,
                                INDArray x,
                                INDArray y,
                                INDArray z,
                                Object[] extraArgs)

Specified by:

createAccum in interface OpFactory

Returns:

createAccum

public Accumulation createAccum(String name,
                                INDArray x,
                                INDArray y)

Specified by:

createAccum in interface OpFactory

Returns:

createIndexAccum

public IndexAccumulation createIndexAccum(String opName,
                                          INDArray x,
                                          INDArray y,
                                          INDArray z,
                                          Object[] extraArgs)

Specified by:

createIndexAccum in interface OpFactory

Parameters:

opName -

x -

y -

z -

extraArgs - @return

createIndexAccum

public IndexAccumulation createIndexAccum(String name,
                                          INDArray x)

Specified by:

createIndexAccum in interface OpFactory

Returns:

createIndexAccum

public IndexAccumulation createIndexAccum(String name,
                                          INDArray x,
                                          INDArray y)

Description copied from interface: OpFactory

Index accumulation operation

Specified by:

createIndexAccum in interface OpFactory

Returns:

createTransform

public TransformOp createTransform(String name,
                                   INDArray x,
                                   INDArray y)

Specified by:

createTransform in interface OpFactory

Returns:

createTransform

public TransformOp createTransform(String name,
                                   INDArray x)

Specified by:

createTransform in interface OpFactory

Returns:

createTransform

public TransformOp createTransform(String name,
                                   INDArray x,
                                   Object[] extraArgs)

Specified by:

createTransform in interface OpFactory

Returns:

createTransform

public TransformOp createTransform(String name,
                                   INDArray x,
                                   INDArray y,
                                   INDArray z)

Specified by:

createTransform in interface OpFactory

Returns:

createTransform

public TransformOp createTransform(String name,
                                   INDArray x,
                                   INDArray y,
                                   INDArray z,
                                   Object[] extraArgs)

Specified by:

createTransform in interface OpFactory

Parameters:

name -

x -

y -

z -

extraArgs -

Returns:

createScalarTransform

public ScalarOp createScalarTransform(String name,
                                      INDArray x,
                                      INDArray y,
                                      double scalar)

Specified by:

createScalarTransform in interface OpFactory

Parameters:

name -

x -

y -

scalar -

Returns:

createScalarTransform

public ScalarOp createScalarTransform(String name,
                                      INDArray x,
                                      double scalar)

Specified by:

createScalarTransform in interface OpFactory

Parameters:

name -

x -

scalar -

Returns:

createScalarTransform

public ScalarOp createScalarTransform(String name,
                                      INDArray x,
                                      Object[] extraArgs,
                                      double scalar)

Specified by:

createScalarTransform in interface OpFactory

Parameters:

name -

x -

extraArgs -

scalar -

Returns:

createScalarTransform

public ScalarOp createScalarTransform(String name,
                                      INDArray x,
                                      INDArray y,
                                      INDArray z,
                                      double scalar)

Specified by:

createScalarTransform in interface OpFactory

Parameters:

name -

x -

y -

z -

scalar -

Returns:

createScalarTransform

public ScalarOp createScalarTransform(String name,
                                      INDArray x,
                                      INDArray y,
                                      INDArray z,
                                      Object[] extraArgs,
                                      double scalar)

Specified by:

createScalarTransform in interface OpFactory

Parameters:

name -

x -

y -

z -

extraArgs -

scalar -

Returns:

createBroadcastOp

public BroadcastOp createBroadcastOp(String name,
                                     INDArray x,
                                     INDArray y,
                                     INDArray z,
                                     int... dimension)

Description copied from interface: OpFactory

Create a vector operation

Specified by:

createBroadcastOp in interface OpFactory

Parameters:

name - Name of the vector op

x - NDArray to operate on

y - Vector

z - Result NDArray

dimension - Dimension to do op along. 0 for row, 1 for column, etc

Returns:

VectorOp

createBroadcastOp

public BroadcastOp createBroadcastOp(String name,
                                     INDArray x,
                                     INDArray y,
                                     INDArray z,
                                     Object[] extraArgs,
                                     int... dimension)

Specified by:

createBroadcastOp in interface OpFactory

Returns:

createBroadcastOp

public BroadcastOp createBroadcastOp(String name,
                                     INDArray x,
                                     INDArray y,
                                     int... dimension)

Description copied from interface: OpFactory

Create a vector operation

Specified by:

createBroadcastOp in interface OpFactory

Parameters:

name - Name of the vector op

x - NDArray to operate on

y - Result NDArray

dimension - Dimension to do op along. 0 for row, 1 for column, etc

Returns:

VectorOp

getOpNumByName

public int getOpNumByName(String opName)

This method returns op id number for given opName

Specified by:

getOpNumByName in interface OpFactory

Parameters:

opName -

Returns:

getOpNumIfExists

public int getOpNumIfExists(String opName)

Description copied from interface: OpFactory

This method returns op id number if opName exists, or -1 otherwise

Specified by:

getOpNumIfExists in interface OpFactory

Returns:

getOpByName

public Op getOpByName(String opName)

Description copied from interface: OpFactory

This method returns Op instance if opName exists, null otherwise

Specified by:

getOpByName in interface OpFactory

Returns: