Multiplies slices of two tensors in batches.
Compat aliases for migration
See Migration guide for more details.
tf.batch_mat_mul_v3( x, y, Tout, adj_x=False, adj_y=False, name=None )
Multiplies all slices of
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
adj_y flag to
True, which are by default
The input tensors
y are 2-D or higher with shape
[..., r_x, c_x]
[..., 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[..., :, :])