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