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.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

static <T extends TNumber> 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

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "BroadcastGradientArgs"

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