org.deeplearning4j.spark.api.worker

Class NetBroadcastTuple

java.lang.Object

org.deeplearning4j.spark.api.worker.NetBroadcastTuple

All Implemented Interfaces:

java.io.Serializable

public class NetBroadcastTuple
extends java.lang.Object
implements java.io.Serializable

A simple class for storing configurations, parameters and updaters in one class (so they can be broadcast together)

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
NetBroadcastTuple(ComputationGraphConfiguration graphConfiguration, org.nd4j.linalg.api.ndarray.INDArray parameters, org.nd4j.linalg.api.ndarray.INDArray updaterState)
NetBroadcastTuple(MultiLayerConfiguration configuration, ComputationGraphConfiguration graphConfiguration, org.nd4j.linalg.api.ndarray.INDArray parameters, org.nd4j.linalg.api.ndarray.INDArray updaterState)
NetBroadcastTuple(MultiLayerConfiguration configuration, ComputationGraphConfiguration graphConfiguration, org.nd4j.linalg.api.ndarray.INDArray parameters, org.nd4j.linalg.api.ndarray.INDArray updaterState, java.util.concurrent.atomic.AtomicInteger counter)
NetBroadcastTuple(MultiLayerConfiguration configuration, org.nd4j.linalg.api.ndarray.INDArray parameters, org.nd4j.linalg.api.ndarray.INDArray updaterState)

Method Summary

Methods inherited from class java.lang.Object

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

Constructor Detail

NetBroadcastTuple

public NetBroadcastTuple(MultiLayerConfiguration configuration,
                         org.nd4j.linalg.api.ndarray.INDArray parameters,
                         org.nd4j.linalg.api.ndarray.INDArray updaterState)

NetBroadcastTuple

public NetBroadcastTuple(ComputationGraphConfiguration graphConfiguration,
                         org.nd4j.linalg.api.ndarray.INDArray parameters,
                         org.nd4j.linalg.api.ndarray.INDArray updaterState)

NetBroadcastTuple

public NetBroadcastTuple(MultiLayerConfiguration configuration,
                         ComputationGraphConfiguration graphConfiguration,
                         org.nd4j.linalg.api.ndarray.INDArray parameters,
                         org.nd4j.linalg.api.ndarray.INDArray updaterState)

NetBroadcastTuple

public NetBroadcastTuple(MultiLayerConfiguration configuration,
                         ComputationGraphConfiguration graphConfiguration,
                         org.nd4j.linalg.api.ndarray.INDArray parameters,
                         org.nd4j.linalg.api.ndarray.INDArray updaterState,
                         java.util.concurrent.atomic.AtomicInteger counter)