org.nd4j.linalg.ops

Class BaseElementWiseOp

java.lang.Object

org.nd4j.linalg.ops.BaseElementWiseOp

All Implemented Interfaces:

ElementWiseOp

Direct Known Subclasses:

Abs, BaseTwoArrayElementWiseOp, EqualTo, Exp, Floor, GreaterThan, HardTanh, Identity, LessThan, Log, Max, MaxOut, Negative, NotEqualTo, Pow, Round, Sigmoid, Sign, Sqrt, Stabilize, Tanh

public abstract class BaseElementWiseOp
extends java.lang.Object
implements ElementWiseOp

Baseline element wise operation so only applyTransformToOrigin has to be implemented. This also handles the ability to perform scalar wise operations vs just a functional transformation

Constructor Summary

Constructors

Constructor and Description
BaseElementWiseOp()

Method Summary

Modifier and Type	Method and Description
void	applyTransformToOrigin(INDArray origin, int i) Apply the transformation at from[i]
void	applyTransformToOrigin(INDArray origin, int i, java.lang.Object valueToApply) Apply the transformation at from[i] using the supplied value
void	exec() Apply the transformation
INDArray	from() The input matrix
java.lang.Object	getFromOrigin(INDArray origin, int i) Get the element at from at index i
static java.util.concurrent.ExecutorService	getThreads()

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.nd4j.linalg.ops.ElementWiseOp

apply

Constructor Detail

BaseElementWiseOp

public BaseElementWiseOp()

Method Detail

getThreads

public static java.util.concurrent.ExecutorService getThreads()

applyTransformToOrigin

public void applyTransformToOrigin(INDArray origin,
                                   int i)

Apply the transformation at from[i]

Specified by:

applyTransformToOrigin in interface ElementWiseOp

Parameters:

i - the index of the element to apply the transform to

origin - the origin ndarray

applyTransformToOrigin

public void applyTransformToOrigin(INDArray origin,
                                   int i,
                                   java.lang.Object valueToApply)

Apply the transformation at from[i] using the supplied value

Specified by:

applyTransformToOrigin in interface ElementWiseOp

Parameters:

origin - the origin ndarray

i - the index of the element to applyTransformToOrigin

valueToApply - the value to apply to the given index

getFromOrigin

public java.lang.Object getFromOrigin(INDArray origin,
                                      int i)

Description copied from interface: ElementWiseOp

Get the element at from at index i

Specified by:

getFromOrigin in interface ElementWiseOp

Parameters:

origin - the origin ndarray

i - the index of the element to retrieve

Returns:

the element at index i

from

public INDArray from()

The input matrix

Specified by:

from in interface ElementWiseOp

Returns:

exec

public void exec()

Apply the transformation

Specified by:

exec in interface ElementWiseOp