BroadcastGradientArgs

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>
create(Scope scope, Operand<T> s0, Operand<T> s1)
Factory method to create a class wrapping a new BroadcastGradientArgs operation.
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

public Output<T> r0 ()

public Output<T> r1 ()