BatchMatMulV2

public final class BatchMatMulV2

Multiplies slices of two tensors in batches.

Multiplies all slices of `Tensor` `x` and `y` (each slice can be viewed as an element of a batch), and arranges the individual results in a single output tensor of the same batch size. Each of the individual slices can optionally be adjointed (to adjoint a matrix means to transpose and conjugate it) before multiplication by setting the `adj_x` or `adj_y` flag to `True`, which are by default `False`.

The input tensors `x` and `y` are 2-D or higher with shape `[..., r_x, c_x]` and `[..., r_y, c_y]`.

The output tensor is 2-D or higher with shape `[..., r_o, c_o]`, where:

r_o = c_x if adj_x else r_x c_o = r_y if adj_y else c_y

It is computed as:

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

NOTE: `BatchMatMulV2` supports broadcasting in the batch dimensions. More about broadcasting [here](http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html).

Nested Classes

class BatchMatMulV2.Options Optional attributes for BatchMatMulV2  

Public Methods

static BatchMatMulV2.Options
adjX(Boolean adjX)
static BatchMatMulV2.Options
adjY(Boolean adjY)
Output<T>
asOutput()
Returns the symbolic handle of a tensor.
static <T> BatchMatMulV2<T>
create(Scope scope, Operand<T> x, Operand<T> y, Options... options)
Factory method to create a class wrapping a new BatchMatMulV2 operation.
static BatchMatMulV2.Options
gradX(Boolean gradX)
static BatchMatMulV2.Options
gradY(Boolean gradY)
Output<T>
output()
3-D or higher with shape `[..., r_o, c_o]`

Inherited Methods

Public Methods

public static BatchMatMulV2.Options adjX (Boolean adjX)

Parameters
adjX If `True`, adjoint the slices of `x`. Defaults to `False`.

public static BatchMatMulV2.Options adjY (Boolean adjY)

Parameters
adjY If `True`, adjoint the slices of `y`. Defaults to `False`.

public Output<T> asOutput ()

Returns the symbolic handle of a tensor.

Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.

public static BatchMatMulV2<T> create (Scope scope, Operand<T> x, Operand<T> y, Options... options)

Factory method to create a class wrapping a new BatchMatMulV2 operation.

Parameters
scope current scope
x 2-D or higher with shape `[..., r_x, c_x]`.
y 2-D or higher with shape `[..., r_y, c_y]`.
options carries optional attributes values
Returns
  • a new instance of BatchMatMulV2

public static BatchMatMulV2.Options gradX (Boolean gradX)

public static BatchMatMulV2.Options gradY (Boolean gradY)

public Output<T> output ()

3-D or higher with shape `[..., r_o, c_o]`