tensorflow::
ops::
BatchMatMulV3
#include <math_ops.h>
Multiplies slices of two tensors in batches.
Summary
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
.
Args:
- scope: A Scope object
-
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.
Optional attributes (see
Attrs
):
-
adj_x: If
True
, adjoint the slices ofx
. Defaults toFalse
. -
adj_y: If
True
, adjoint the slices ofy
. Defaults toFalse
.
Returns:
-
Output
: 3-D or higher with shape[..., r_o, c_o]
Constructors and Destructors |
|
---|---|
BatchMatMulV3
(const ::
tensorflow::Scope
& scope, ::
tensorflow::Input
x, ::
tensorflow::Input
y, DataType Tout)
|
|
BatchMatMulV3
(const ::
tensorflow::Scope
& scope, ::
tensorflow::Input
x, ::
tensorflow::Input
y, DataType Tout, const
BatchMatMulV3::Attrs
& attrs)
|
Public attributes |
|
---|---|
operation
|
|
output
|
Public functions |
|
---|---|
node
() const
|
::tensorflow::Node *
|
operator::tensorflow::Input
() const
|
|
operator::tensorflow::Output
() const
|
|
Public static functions |
|
---|---|
AdjX
(bool x)
|
|
AdjY
(bool x)
|
Structs |
|
---|---|
tensorflow::
|
Optional attribute setters for BatchMatMulV3 . |
Public attributes
Public functions
BatchMatMulV3
BatchMatMulV3( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input y, DataType Tout )
BatchMatMulV3
BatchMatMulV3( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input y, DataType Tout, const BatchMatMulV3::Attrs & attrs )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const