public final class
BroadcastGradientArgs
Return the reduction indices for computing gradients of s0 op s1 with broadcast.
This is typically used by gradient computations for a broadcasting operation.
Public Methods
| static <T extends Number> BroadcastGradientArgs <T> | |
| Output <T> |
r0
()
|
| Output <T> |
r1
()
|
Inherited Methods
Public Methods
public static BroadcastGradientArgs <T> create ( Scope scope, Operand <T> s0, Operand <T> s1)
Factory method to create a class wrapping a new BroadcastGradientArgs operation.
Parameters
| scope | current scope |
|---|
Returns
- a new instance of BroadcastGradientArgs