org.nd4j.linalg.api.ops.factory

Interface OpFactory

All Known Implementing Classes:

DefaultOpFactory

public interface OpFactory

Op factory

Author:

Adam Gibson

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

Method Detail

createGradientOp

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

createShape

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

Parameters:

name -

x -

z -

extraArgs -

Returns:

createLossFunction

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

Create a loss function with the given inputs and outputs

Parameters:

name - the opName of the function

x - the input

y - the output

Returns:

a loss function representing the delta between the 2

createAccum

Accumulation createAccum(String name,
                         INDArray x)

Accumulation operation

Parameters:

name - the opName of the function to create

x - the input to the function

Returns:

the operation

createAccum

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

Accumulation operation

Parameters:

name - the opName of the function

x - the input

y - the pairwise transformation

z - the output

Returns:

the operation

createAccum

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

Parameters:

name -

x -

y -

z -

extraArgs -

Returns:

createAccum

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

Parameters:

name -

x -

y -

Returns:

createIndexAccum

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

Parameters:

opName -

x -

y -

z -

extraArgs - @return

createIndexAccum

IndexAccumulation createIndexAccum(String name,
                                   INDArray x)

Parameters:

name -

x -

Returns:

createIndexAccum

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

Index accumulation operation

Parameters:

name -

x -

y -

Returns:

createTransform

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

Parameters:

name -

x -

y -

Returns:

createTransform

TransformOp createTransform(String name,
                            INDArray x)

Parameters:

name -

x -

Returns:

createTransform

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

Parameters:

name -

x -

extraArgs -

Returns:

createTransform

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

Parameters:

name -

x -

y -

z -

Returns:

createTransform

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

Parameters:

name -

x -

y -

z -

createScalarTransform

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

Parameters:

name -

x -

y -

scalar -

Returns:

createScalarTransform

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

Parameters:

name -

x -

scalar -

Returns:

createScalarTransform

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

Parameters:

name -

x -

extraArgs -

scalar -

Returns:

createScalarTransform

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

Parameters:

name -

x -

y -

z -

scalar -

Returns:

createScalarTransform

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

Parameters:

name -

x -

y -

z -

scalar -

Returns:

createBroadcastOp

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

Create a vector operation

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

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

Parameters:

name -

x -

y -

z -

extraArgs -

dimension -

Returns:

createBroadcastOp

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

Create a vector operation

Parameters:

name - Name of the vector op

x - NDArray to operate on

z - Result NDArray

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

Returns:

VectorOp

getOpNumByName

int getOpNumByName(String opName)

This method returns op id number for given opName

Returns:

getOpNumIfExists

int getOpNumIfExists(String opName)

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

Parameters:

opName -

Returns:

getOpByName

Op getOpByName(String opName)

This method returns Op instance if opName exists, null otherwise

Parameters:

opName -

Returns: