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, Negative, Pow, Round, Sigmoid, Sqrt, Stabilize, Tanh

public abstract class BaseElementWiseOp
extends 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

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
protected static ExecutorService	dimensionThreads
protected INDArray	from
protected Object	scalarValue

Constructor Summary

Constructors

Constructor and Description
BaseElementWiseOp()

Method Summary

Methods

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

Methods inherited from class java.lang.Object

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

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

apply

Field Detail

from

protected INDArray from

scalarValue

protected Object scalarValue

dimensionThreads

protected static ExecutorService dimensionThreads

Constructor Detail

BaseElementWiseOp

public BaseElementWiseOp()

Method Detail

getThreads

public static 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 applyTransformToOrigin

origin - the origin ndarray

applyTransformToOrigin

public void applyTransformToOrigin(INDArray origin,
                          int i,
                          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 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