MatMul

public final class MatMul

Multiply the matrix "a" by the matrix "b".

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.

Nested Classes

class MatMul.Options Optional attributes for MatMul  

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

Output<T>
asOutput()
Returns the symbolic handle of the tensor.
static <T extends TType> MatMul<T>
create(Scope scope, Operand<T> a, Operand<T> b, Options... options)
Factory method to create a class wrapping a new MatMul operation.
Output<T>
static MatMul.Options
transposeA(Boolean transposeA)
static MatMul.Options
transposeB(Boolean transposeB)

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "MatMul"

Public Methods

public Output<T> asOutput ()

Returns the symbolic handle of the 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 MatMul<T> create (Scope scope, Operand<T> a, Operand<T> b, Options... options)

Factory method to create a class wrapping a new MatMul operation.

Parameters
scope current scope
options carries optional attributes values
Returns
  • a new instance of MatMul

public Output<T> product ()

public static MatMul.Options transposeA (Boolean transposeA)

Parameters
transposeA If true, "a" is transposed before multiplication.

public static MatMul.Options transposeB (Boolean transposeB)

Parameters
transposeB If true, "b" is transposed before multiplication.