Class BaseAggregate
- java.lang.Object
-
- org.nd4j.linalg.api.ops.aggregates.BaseAggregate
-
- All Implemented Interfaces:
Aggregate
- Direct Known Subclasses:
AggregateAxpy,AggregateGEMM
public abstract class BaseAggregate extends Object implements Aggregate
-
-
Field Summary
Fields Modifier and Type Field Description protected List<INDArray>argumentsprotected NumberfinalResultprotected List<Integer>indexingArgumentsprotected List<int[]>intArrayArgumentsprotected List<Number>realArgumentsprotected List<DataBuffer>shapes
-
Constructor Summary
Constructors Constructor Description BaseAggregate()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description List<INDArray>getArguments()NumbergetFinalResult()List<Integer>getIndexingArguments()List<int[]>getIntArrayArguments()List<Number>getRealArguments()longgetRequiredBatchMemorySize()This method returns amount of memory required for batch creation for this specific AggregateList<DataBuffer>getShapes()voidsetFinalResult(Number result)-
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.api.ops.aggregates.Aggregate
getSharedMemorySize, getThreadsPerInstance, maxArguments, maxIndexArguments, maxIntArrays, maxIntArraySize, maxRealArguments, maxShapes, name, opNum
-
-
-
-
Field Detail
-
shapes
protected List<DataBuffer> shapes
-
intArrayArguments
protected List<int[]> intArrayArguments
-
finalResult
protected Number finalResult
-
-
Method Detail
-
getArguments
public List<INDArray> getArguments()
- Specified by:
getArgumentsin interfaceAggregate- Returns:
-
getFinalResult
public Number getFinalResult()
- Specified by:
getFinalResultin interfaceAggregate- Returns:
-
setFinalResult
public void setFinalResult(Number result)
- Specified by:
setFinalResultin interfaceAggregate
-
getShapes
public List<DataBuffer> getShapes()
-
getIndexingArguments
public List<Integer> getIndexingArguments()
- Specified by:
getIndexingArgumentsin interfaceAggregate- Returns:
-
getRealArguments
public List<Number> getRealArguments()
- Specified by:
getRealArgumentsin interfaceAggregate- Returns:
-
getIntArrayArguments
public List<int[]> getIntArrayArguments()
- Specified by:
getIntArrayArgumentsin interfaceAggregate- Returns:
-
getRequiredBatchMemorySize
public long getRequiredBatchMemorySize()
Description copied from interface:AggregateThis method returns amount of memory required for batch creation for this specific Aggregate- Specified by:
getRequiredBatchMemorySizein interfaceAggregate- Returns:
-
-