Package 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 Description PublishingListener()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description voidonNDArray(org.nd4j.linalg.api.ndarray.INDArray arr)Setup an ndarrayvoidonNDArrayMessage(NDArrayMessage message)A listener for ndarray messagevoidonNDArrayPartial(org.nd4j.linalg.api.ndarray.INDArray arr, long idx, int... dimensions)Used for partial updates using tensor along dimension
-
-
-
Method Detail
-
onNDArrayMessage
public void onNDArrayMessage(NDArrayMessage message)
A listener for ndarray message- Specified by:
onNDArrayMessagein interfaceNDArrayCallback- 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:
onNDArrayPartialin interfaceNDArrayCallback- Parameters:
arr- the array to count as an updateidx- the index for the tensor along dimensiondimensions- 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:
onNDArrayin interfaceNDArrayCallback- Parameters:
arr-
-
-