com.thoughtworks.deeplearning.plugins
A plugin that provides differentiable operators on neural networks whose Data and Delta is scala.Float.
杨博 (Yang Bo)
Unlike FloatLayers, FloatLayer in this CumulativeFloatLayers will share Tapes created in forward pass pass for all dependencies, avoiding re-evaluation in the case of diamond dependencies in a neural network.
CumulativeFloatLayers
A plugin that provides differentiable operators on neural networks whose Data and Delta is scala.Float.
Author:
杨博 (Yang Bo)
Unlike FloatLayers, FloatLayer in this
CumulativeFloatLayers
will share Tapes created in forward pass pass for all dependencies, avoiding re-evaluation in the case of diamond dependencies in a neural network.