org.tensorflow.op.core

Class BroadcastGradientArgs<T extends Number>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.BroadcastGradientArgs<T>

Type Parameters:

T - data type for r0() output

All Implemented Interfaces:

Op

public final class BroadcastGradientArgs<T extends Number>
extends PrimitiveOp

Return the reduction indices for computing gradients of s0 op s1 with broadcast.

This is typically used by gradient computations for a broadcasting operation.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static <T extends Number> BroadcastGradientArgs<T>	create(Scope scope, Operand<T> s0, Operand<T> s1) Factory method to create a class to wrap a new BroadcastGradientArgs operation to the graph.
Output<T>	r0()
Output<T>	r1()

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static <T extends Number> BroadcastGradientArgs<T> create(Scope scope,
                                                                 Operand<T> s0,
                                                                 Operand<T> s1)

Factory method to create a class to wrap a new BroadcastGradientArgs operation to the graph.

Parameters:

scope - current graph scope

s0 -

s1 -

Returns:

a new instance of BroadcastGradientArgs

r0

public Output<T> r0()

r1

public Output<T> r1()