org.nd4j.parameterserver.updater.storage

Class InMemoryUpdateStorage

java.lang.Object

org.nd4j.parameterserver.updater.storage.BaseUpdateStorage

org.nd4j.parameterserver.updater.storage.InMemoryUpdateStorage

All Implemented Interfaces:

UpdateStorage

public class InMemoryUpdateStorage
extends BaseUpdateStorage

Constructor Summary

Constructors

Constructor and Description
InMemoryUpdateStorage()

Method Summary

Modifier and Type	Method and Description
void	addUpdate(NDArrayMessage array) Add an ndarray to the storage
void	clear() Clear the array storage
NDArrayMessage	doGetUpdate(int index) A method for actually performing the implementation of retrieving the ndarray
int	numUpdates() The number of updates added to the update storage

Methods inherited from class org.nd4j.parameterserver.updater.storage.BaseUpdateStorage

close, getUpdate

Methods inherited from class java.lang.Object

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

Constructor Detail

InMemoryUpdateStorage

public InMemoryUpdateStorage()

Method Detail

addUpdate

public void addUpdate(NDArrayMessage array)

Add an ndarray to the storage

Parameters:

array - the array to add

numUpdates

public int numUpdates()

The number of updates added to the update storage

Returns:

clear

public void clear()

Clear the array storage

doGetUpdate

public NDArrayMessage doGetUpdate(int index)

A method for actually performing the implementation of retrieving the ndarray

Specified by:

doGetUpdate in class BaseUpdateStorage

Parameters:

index - the index of the INDArray to get

Returns:

the ndarray at the specified index