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 and Description
OptimizedGraphArrayHolder(ArrayHolder underlyingHolder)

Method Summary

Modifier and Type	Method and 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 name
void	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()

Methods inherited from class java.lang.Object

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

Constructor Detail

OptimizedGraphArrayHolder

public OptimizedGraphArrayHolder(ArrayHolder underlyingHolder)

Method Detail

setFunction

public void setFunction(String name,
                        Supplier<INDArray> fn)

hasArray

public boolean hasArray(String name)

Specified by:

hasArray in interface ArrayHolder

Returns:

True if an array by that name exists

getArray

public INDArray getArray(String name)

Specified by:

getArray in interface ArrayHolder

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 interface ArrayHolder

Parameters:

name - Name of the array

array - 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 interface ArrayHolder

Parameters:

name - Name of the array to return

Returns:

The now-removed array

size

public int size()

Specified by:

size in interface ArrayHolder

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 interface ArrayHolder

Parameters:

arrayHolder - Array holder to initialize this based on

arrayNames

public Collection<String> arrayNames()

Specified by:

arrayNames in interface ArrayHolder

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 interface ArrayHolder

Parameters:

from - Original name

to - New name