Package org.nd4j.autodiff.samediff.array
Class OptimizedGraphArrayHolder
- java.lang.Object
-
- org.nd4j.autodiff.samediff.array.OptimizedGraphArrayHolder
-
- All Implemented Interfaces:
ArrayHolder
public class OptimizedGraphArrayHolder extends Object implements ArrayHolder
-
-
Constructor Summary
Constructors Constructor Description OptimizedGraphArrayHolder(ArrayHolder underlyingHolder)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Collection<String>
arrayNames()
INDArray
getArray(String name)
boolean
hasArray(String name)
void
initFrom(ArrayHolder arrayHolder)
Initialize from the specified array holder.INDArray
removeArray(String name)
Remove the array from the ArrayHolder, returning it (if it exists)void
rename(String from, String to)
Rename the entry with the specified namevoid
setArray(String name, INDArray array)
Set the array for the specified name (new array, or replace if it already exists)void
setFunction(String name, Supplier<INDArray> fn)
int
size()
-
-
-
Constructor Detail
-
OptimizedGraphArrayHolder
public OptimizedGraphArrayHolder(ArrayHolder underlyingHolder)
-
-
Method Detail
-
hasArray
public boolean hasArray(String name)
- Specified by:
hasArray
in interfaceArrayHolder
- Returns:
- True if an array by that name exists
-
getArray
public INDArray getArray(String name)
- Specified by:
getArray
in interfaceArrayHolder
- Parameters:
name
- Name of the array to get- Returns:
- The array, or null if no array with that name exists
-
setArray
public void setArray(String name, INDArray array)
Description copied from interface:ArrayHolder
Set the array for the specified name (new array, or replace if it already exists)- Specified by:
setArray
in interfaceArrayHolder
- Parameters:
name
- Name of the arrayarray
- Array to set
-
removeArray
public INDArray removeArray(String name)
Description copied from interface:ArrayHolder
Remove the array from the ArrayHolder, returning it (if it exists)- Specified by:
removeArray
in interfaceArrayHolder
- Parameters:
name
- Name of the array to return- Returns:
- The now-removed array
-
size
public int size()
- Specified by:
size
in interfaceArrayHolder
- Returns:
- Number of arrays in the ArrayHolder
-
initFrom
public void initFrom(ArrayHolder arrayHolder)
Description copied from interface:ArrayHolder
Initialize from the specified array holder. This clears all internal arrays, and adds all arrays from the specified array holder- Specified by:
initFrom
in interfaceArrayHolder
- Parameters:
arrayHolder
- Array holder to initialize this based on
-
arrayNames
public Collection<String> arrayNames()
- Specified by:
arrayNames
in interfaceArrayHolder
- Returns:
- Names of the arrays currently in the ArrayHolder
-
rename
public void rename(String from, String to)
Description copied from interface:ArrayHolder
Rename the entry with the specified name- Specified by:
rename
in interfaceArrayHolder
- Parameters:
from
- Original nameto
- New name
-
-