BroadcastHelper

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>
create ( Scope scope, Operand <T> lhs, Operand <T> rhs, Operand <? extends TNumber > broadcastDims)
Factory method to create a class wrapping a new BroadcastHelper operation.
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

public Output <T> lhsOutput ()

the broadcasted LHS tensor

public Output <T> rhsOutput ()

the broadcasted RHS tensor