org.nd4j.autodiff.samediff.array

Class ThreadSafeArrayHolder

java.lang.Object

org.nd4j.autodiff.samediff.array.ThreadSafeArrayHolder

All Implemented Interfaces:

ArrayHolder

public class ThreadSafeArrayHolder
extends Object
implements ArrayHolder

An ArrayHolder that uses the thread safe DeviceLocalNDArray internally

Author:

Alex Black

Constructor Summary

Constructors

Constructor and Description
ThreadSafeArrayHolder(boolean lazyInit)

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)
int	size()

Methods inherited from class java.lang.Object

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

Constructor Detail

ThreadSafeArrayHolder

public ThreadSafeArrayHolder(boolean lazyInit)

Parameters:

lazyInit - If true: use lazy initialization for DeviceLocalNDArray

Method Detail

hasArray

public boolean hasArray(@NonNull
                        String name)

Specified by:

hasArray in interface ArrayHolder

Returns:

True if an array by that name exists

getArray

public INDArray getArray(@NonNull
                         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(@NonNull
                     String name,
                     @NonNull
                     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(@NonNull
                            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(@NonNull
                   String from,
                   @NonNull
                   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