org.nd4j.linalg.ops

Class BaseTwoArrayElementWiseOp

java.lang.Object

org.nd4j.linalg.ops.BaseElementWiseOp

org.nd4j.linalg.ops.BaseTwoArrayElementWiseOp

All Implemented Interfaces:

ElementWiseOp, TwoArrayElementWiseOp

Direct Known Subclasses:

AddOp, DivideOp, MultiplyOp, SubtractOp

public abstract class BaseTwoArrayElementWiseOp
extends BaseElementWiseOp
implements TwoArrayElementWiseOp

Apply an operation and save it to a resulting matrix

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
protected INDArray	other
protected INDArray	to

Fields inherited from class org.nd4j.linalg.ops.BaseElementWiseOp

dimensionThreads, from, scalarValue

Constructor Summary

Constructors

Constructor and Description
BaseTwoArrayElementWiseOp()

Method Summary

Methods

Modifier and Type	Method and Description
Object	apply(INDArray origin, Object value, int i) The transformation for a given value
void	applyTransformToDestination(INDArray from, INDArray destination, INDArray other, int i) Apply the function from to the specified index in to.
protected abstract IComplexNumber	complexComplex(IComplexNumber num1, IComplexNumber num2)
protected abstract IComplexNumber	complexReal(IComplexNumber origin, float secondValue)
protected Object	doOp(INDArray originNDArray, int i, Object value)
void	exec() Executes the operation across the matrix
Object	getOther(INDArray other, int i) Returns the element in destination at index i
protected abstract IComplexNumber	realComplex(float real, IComplexNumber other)
protected abstract float	realReal(float firstElement, float secondElement)
INDArray	to() The output matrix

Methods inherited from class org.nd4j.linalg.ops.BaseElementWiseOp

applyTransformToOrigin, applyTransformToOrigin, from, getFromOrigin, 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

applyTransformToOrigin, applyTransformToOrigin, from, getFromOrigin

Field Detail

to

protected INDArray to

other

protected INDArray other

Constructor Detail

BaseTwoArrayElementWiseOp

public BaseTwoArrayElementWiseOp()

Method Detail

applyTransformToDestination

public void applyTransformToDestination(INDArray from,
                               INDArray destination,
                               INDArray other,
                               int i)

Apply the function from to the specified index in to. The value from to is passed in to apply and then a transform of the matching elements in both from and to are used for a transformation. If a scalar is specified, this will apply a scalar wise operation based on the scalar and the origin matrix instead

Specified by:

applyTransformToDestination in interface TwoArrayElementWiseOp

Parameters:

i - the index to apply to

from - the origin ndarray

destination - the destination ndarray

other - the other ndarray

exec

public void exec()

Executes the operation across the matrix

Specified by:

exec in interface ElementWiseOp

Specified by:

exec in interface TwoArrayElementWiseOp

Overrides:

exec in class BaseElementWiseOp

getOther

public Object getOther(INDArray other,
              int i)

Returns the element in destination at index i

Specified by:

getOther in interface TwoArrayElementWiseOp

Parameters:

i - the index of the element to retrieve

other - the destination ndarray

Returns:

the element at index i

doOp

protected Object doOp(INDArray originNDArray,
          int i,
          Object value)

complexComplex

protected abstract IComplexNumber complexComplex(IComplexNumber num1,
                            IComplexNumber num2)

realComplex

protected abstract IComplexNumber realComplex(float real,
                         IComplexNumber other)

complexReal

protected abstract IComplexNumber complexReal(IComplexNumber origin,
                         float secondValue)

realReal

protected abstract float realReal(float firstElement,
             float secondElement)

apply

public Object apply(INDArray origin,
           Object value,
           int i)

The transformation for a given value

Specified by:

apply in interface ElementWiseOp

Parameters:

value - the value to applyTransformToOrigin

origin - the origin ndarray

i - the index of the element being acted upon

Returns:

the transformed value based on the input

to

public INDArray to()

The output matrix

Specified by:

to in interface TwoArrayElementWiseOp

Returns: