public class AutoDiffBroadcast
extends java.lang.Object
Constructor and Description |
---|
AutoDiffBroadcast() |
Modifier and Type | Method and Description |
---|---|
static PartialDerivative |
correctForBroadcastPartialForward(PartialDerivative partial,
long[] partialOfShape,
long[] targetOfShape) |
static PartialDerivative |
correctForBroadcastPartialReverse(PartialDerivative partial,
long[] partialWrtShape,
long[] targetWrtShape) |
public static PartialDerivative correctForBroadcastPartialForward(PartialDerivative partial, long[] partialOfShape, long[] targetOfShape)
public static PartialDerivative correctForBroadcastPartialReverse(PartialDerivative partial, long[] partialWrtShape, long[] targetWrtShape)