BatchMatMulV3

public final class BatchMatMulV3

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: `BatchMatMulV3` supports broadcasting in the batch dimensions. More about broadcasting [here](http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html).

Nested Classes

class BatchMatMulV3.Options Optional attributes for BatchMatMulV3  

Public Methods

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

Inherited Methods

Public Methods

public static BatchMatMulV3.Options adjX (Boolean adjX)

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

public static BatchMatMulV3.Options adjY (Boolean adjY)

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

public Output<V> 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 BatchMatMulV3<V> create (Scope scope, Operand<T> x, Operand<U> y, Class<V> Tout, Options... options)

Factory method to create a class wrapping a new BatchMatMulV3 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]`.
Tout If not spcified, Tout is the same type to input type.
options carries optional attributes values
Returns
  • a new instance of BatchMatMulV3

public static BatchMatMulV3.Options gradX (Boolean gradX)

public static BatchMatMulV3.Options gradY (Boolean gradY)

public Output<V> output ()

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