public final class
BroadcastHelper
Helper operator for performing XLA-style broadcasts
Broadcasts `lhs` and `rhs` to the same rank, by adding size 1 dimensions to whichever of `lhs` and `rhs` has the lower rank, using XLA's broadcasting rules for binary operators.
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
static <T extends TType> BroadcastHelper<T> | |
Output<T> |
lhsOutput()
the broadcasted LHS tensor
|
Output<T> |
rhsOutput()
the broadcasted RHS tensor
|
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Constant Value:
"XlaBroadcastHelper"
Public Methods
public static BroadcastHelper<T> create (Scope scope, Operand<T> lhs, Operand<T> rhs, Operand<? extends TNumber> broadcastDims)
Factory method to create a class wrapping a new BroadcastHelper operation.
Parameters
scope | current scope |
---|---|
lhs | the LHS input tensor |
rhs | the RHS input tensor |
broadcastDims | an XLA-style broadcast dimension specification |
Returns
- a new instance of BroadcastHelper