ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

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 ()