T
- data type for product()
output@Operator(group="linalg") public final class MatMul<T> extends PrimitiveOp implements Operand<T>
The inputs must be two-dimensional matrices and the inner dimension of "a" (after being transposed if transpose_a is true) must match the outer dimension of "b" (after being transposed if transposed_b is true).
Note: The default kernel implementation for MatMul on GPUs uses cublas.
Modifier and Type | Class and Description |
---|---|
static class |
MatMul.Options
Optional attributes for
MatMul |
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T> MatMul<T> |
create(Scope scope,
Operand<T> a,
Operand<T> b,
MatMul.Options... options)
Factory method to create a class wrapping a new MatMul operation.
|
Output<T> |
product() |
static MatMul.Options |
transposeA(Boolean transposeA) |
static MatMul.Options |
transposeB(Boolean transposeB) |
equals, hashCode, op, toString
public static <T> MatMul<T> create(Scope scope, Operand<T> a, Operand<T> b, MatMul.Options... options)
scope
- current scopea
- b
- options
- carries optional attributes valuespublic static MatMul.Options transposeA(Boolean transposeA)
transposeA
- If true, "a" is transposed before multiplication.public static MatMul.Options transposeB(Boolean transposeB)
transposeB
- If true, "b" is transposed before multiplication.public Output<T> asOutput()
Operand
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.
asOutput
in interface Operand<T>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.