Constructor and Description |
---|
DefaultOpFactory() |
public LossFunction createLossFunction(String name, INDArray x, INDArray y)
OpFactory
createLossFunction
in interface OpFactory
name
- the name of the functionx
- the inputy
- the outputpublic Accumulation createAccum(String name, INDArray x)
OpFactory
createAccum
in interface OpFactory
name
- the name of the function to createx
- the input to the functionpublic Accumulation createAccum(String name, INDArray x, INDArray y, INDArray z)
OpFactory
createAccum
in interface OpFactory
name
- the name of the functionx
- the inputy
- the pairwise transformationz
- the outputpublic Accumulation createAccum(String name, INDArray x, INDArray y)
createAccum
in interface OpFactory
public IndexAccumulation createIndexAccum(String name, INDArray x)
OpFactory
createIndexAccum
in interface OpFactory
name
- the name of the function to createx
- the input to the functionpublic IndexAccumulation createIndexAccum(String name, INDArray x, INDArray y)
OpFactory
createIndexAccum
in interface OpFactory
public TransformOp createTransform(String name, INDArray x, INDArray y)
createTransform
in interface OpFactory
public TransformOp createTransform(String name, INDArray x)
createTransform
in interface OpFactory
public TransformOp createTransform(String name, INDArray x, Object[] extraArgs)
createTransform
in interface OpFactory
public TransformOp createTransform(String name, INDArray x, INDArray y, INDArray z)
createTransform
in interface OpFactory
public BroadcastOp createBroadcastOp(String name, INDArray x, INDArray y, INDArray z, int... dimension)
OpFactory
createBroadcastOp
in interface OpFactory
name
- Name of the vector opx
- NDArray to operate ony
- Vectorz
- Result NDArraydimension
- Dimension to do op along. 0 for row, 1 for column, etcpublic BroadcastOp createBroadcastOp(String name, INDArray x, INDArray y, int... dimension)
OpFactory
createBroadcastOp
in interface OpFactory
name
- Name of the vector opx
- NDArray to operate ony
- Result NDArraydimension
- Dimension to do op along. 0 for row, 1 for column, etcCopyright © 2016. All Rights Reserved.