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