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>
arguments
protected Number
finalResult
protected List<Integer>
indexingArguments
protected List<int[]>
intArrayArguments
protected List<Number>
realArguments
protected 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()
Number
getFinalResult()
List<Integer>
getIndexingArguments()
List<int[]>
getIntArrayArguments()
List<Number>
getRealArguments()
long
getRequiredBatchMemorySize()
This method returns amount of memory required for batch creation for this specific AggregateList<DataBuffer>
getShapes()
void
setFinalResult(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:
getArguments
in interfaceAggregate
- Returns:
-
getFinalResult
public Number getFinalResult()
- Specified by:
getFinalResult
in interfaceAggregate
- Returns:
-
setFinalResult
public void setFinalResult(Number result)
- Specified by:
setFinalResult
in interfaceAggregate
-
getShapes
public List<DataBuffer> getShapes()
-
getIndexingArguments
public List<Integer> getIndexingArguments()
- Specified by:
getIndexingArguments
in interfaceAggregate
- Returns:
-
getRealArguments
public List<Number> getRealArguments()
- Specified by:
getRealArguments
in interfaceAggregate
- Returns:
-
getIntArrayArguments
public List<int[]> getIntArrayArguments()
- Specified by:
getIntArrayArguments
in interfaceAggregate
- Returns:
-
getRequiredBatchMemorySize
public long getRequiredBatchMemorySize()
Description copied from interface:Aggregate
This method returns amount of memory required for batch creation for this specific Aggregate- Specified by:
getRequiredBatchMemorySize
in interfaceAggregate
- Returns:
-
-