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
BroadcastOp	createBroadcastOp(String name, INDArray x, INDArray y, INDArray z, int... dimension) Create a vector operation
BroadcastOp	createBroadcastOp(String name, INDArray x, INDArray y, int... dimension) Create a vector operation
IndexAccumulation	createIndexAccum(String name, INDArray x) Index accumulation operation
IndexAccumulation	createIndexAccum(String name, INDArray x, INDArray y) Index accumulation operation
LossFunction	createLossFunction(String name, INDArray x, INDArray y) Create a loss function with the given inputs and outputs
TransformOp	createTransform(String name, INDArray x)
TransformOp	createTransform(String name, INDArray x, INDArray y)
TransformOp	createTransform(String name, INDArray x, INDArray y, INDArray z)
protected Class<? extends Op>	lookupFunctionByName(String name)

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

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)

Specified by:

createAccum in interface OpFactory

Returns:

createIndexAccum

public IndexAccumulation createIndexAccum(String name,
                                          INDArray x)

Description copied from interface: OpFactory

Index accumulation operation

Specified by:

createIndexAccum in interface OpFactory

Parameters:

name - the name of the function to create

x - the input to the function

Returns:

the operation

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,
                                   INDArray y,
                                   INDArray z)

Specified by:

createTransform in interface OpFactory

Returns:

lookupFunctionByName

protected Class<? extends Op> lookupFunctionByName(String name)

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