@Namespace(value="tensorflow::ops") @NoOffset @Properties(inherit=tensorflow.class) public class BatchMatMul extends Pointer
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[..., :, :])
Arguments:
* 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]
.
Optional attributes (see Attrs
):
* adj_x: If True
, adjoint the slices of x
. Defaults to False
.
* adj_y: If True
, adjoint the slices of y
. Defaults to False
.
Returns:
* Output
: 3-D or higher with shape [..., r_o, c_o]
Modifier and Type | Class and Description |
---|---|
static class |
BatchMatMul.Attrs
Optional attribute setters for BatchMatMul
|
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
BatchMatMul(Pointer p)
Pointer cast constructor.
|
BatchMatMul(Scope scope,
Input x,
Input y) |
BatchMatMul(Scope scope,
Input x,
Input y,
BatchMatMul.Attrs attrs) |
Modifier and Type | Method and Description |
---|---|
static BatchMatMul.Attrs |
AdjX(boolean x) |
static BatchMatMul.Attrs |
AdjY(boolean x) |
Input |
asInput() |
Output |
asOutput() |
Node |
node() |
Operation |
operation() |
BatchMatMul |
operation(Operation setter) |
Output |
output() |
BatchMatMul |
output(Output setter) |
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public BatchMatMul(Pointer p)
Pointer(Pointer)
.public Node node()
@ByVal public static BatchMatMul.Attrs AdjX(@Cast(value="bool") boolean x)
@ByVal public static BatchMatMul.Attrs AdjY(@Cast(value="bool") boolean x)
public BatchMatMul operation(Operation setter)
public BatchMatMul output(Output setter)
Copyright © 2022. All rights reserved.