org.nd4j.parameterserver

Class PublishingListener

java.lang.Object

org.nd4j.parameterserver.PublishingListener

All Implemented Interfaces:

NDArrayCallback

public class PublishingListener
extends Object
implements NDArrayCallback

Publishing listener for publishing to a master url.

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
PublishingListener()

Method Summary

Modifier and Type	Method and Description
void	onNDArray(org.nd4j.linalg.api.ndarray.INDArray arr) Setup an ndarray
void	onNDArrayMessage(NDArrayMessage message) A listener for ndarray message
void	onNDArrayPartial(org.nd4j.linalg.api.ndarray.INDArray arr, long idx, int... dimensions) Used for partial updates using tensor along dimension

Methods inherited from class java.lang.Object

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

Constructor Detail

PublishingListener

public PublishingListener()

Method Detail

onNDArrayMessage

public void onNDArrayMessage(NDArrayMessage message)

A listener for ndarray message

Specified by:

onNDArrayMessage in interface NDArrayCallback

Parameters:

message - the message for the callback

onNDArrayPartial

public void onNDArrayPartial(org.nd4j.linalg.api.ndarray.INDArray arr,
                             long idx,
                             int... dimensions)

Used for partial updates using tensor along dimension

Specified by:

onNDArrayPartial in interface NDArrayCallback

Parameters:

arr - the array to count as an update

idx - the index for the tensor along dimension

dimensions - the dimensions to act on for the tensor along dimension

onNDArray

public void onNDArray(org.nd4j.linalg.api.ndarray.INDArray arr)

Setup an ndarray

Specified by:

onNDArray in interface NDArrayCallback

Parameters:

arr -