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

Method Detail

createShape

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

Parameters:

name -

x -

z -

Returns:

createLossFunction

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

Create a loss function with the given inputs and outputs

Parameters:

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