Overview > klyn.math.nn.layers > Flatten > differentiate
Version 0.2.1 Overview Index Tree Deprecated
klyn.math.nn.layers.Flatten.differentiate
method
public override differentiate(inputs as NDArray<T>, upstream as NDArray<T>) as LayerGradients<T>
Description

Restores the original sample axes in the gradient.

Parameters
ParameterDescription
inputsOriginal unflattened batch.
upstreamFlattened derivative.
Returns

Same-shaped input gradient with no parameter derivatives.

Throws
  • ValueException if flattened gradient dimensions disagree.
Example

gradient = flatten.differentiate(images, upstream)