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
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)
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)

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

createShape

public Op createShape(String name,
                      INDArray x,
                      INDArray z)

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 name 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 name 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 name 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