SparseMatrixMatMul

public final class SparseMatrixMatMul

Matrix-multiplies a sparse matrix with a dense matrix.

Returns a dense matrix. For inputs A and B, where A is CSR and B is dense; this op returns a dense C;

If transpose_output is false, returns: 

C = A . B
If transpose_output is `true`, returns: 
C = transpose(A . B) = transpose(B) . transpose(A)
where the transposition is performed along the two innermost (matrix) dimensions.

If conjugate_output is `true`, returns: 

C = conjugate(A . B) = conjugate(A) . conjugate(B)
If both conjugate_output and transpose_output are `true`, returns: 
C = conjugate(transpose(A . B)) = conjugate(transpose(B)) .
                                     conjugate(transpose(A))

Nested Classes

class SparseMatrixMatMul.Options Optional attributes for SparseMatrixMatMul  

Constants

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

Public Methods

static SparseMatrixMatMul.Options
adjointA(Boolean adjointA)
static SparseMatrixMatMul.Options
adjointB(Boolean adjointB)
Output<T>
asOutput()
Returns the symbolic handle of the tensor.
static SparseMatrixMatMul.Options
conjugateOutput(Boolean conjugateOutput)
static <T extends TType> SparseMatrixMatMul<T>
create(Scope scope, Operand<?> a, Operand<T> b, Options... options)
Factory method to create a class wrapping a new SparseMatrixMatMul operation.
Output<T>
output()
A dense output tensor.
static SparseMatrixMatMul.Options
transposeA(Boolean transposeA)
static SparseMatrixMatMul.Options
transposeB(Boolean transposeB)
static SparseMatrixMatMul.Options
transposeOutput(Boolean transposeOutput)

Inherited Methods

org.tensorflow.op.RawOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Return this unit of computation as a single Operation.
final String
toString()
boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()
org.tensorflow.op.Op
abstract ExecutionEnvironment
env()
Return the execution environment this op was created in.
abstract Operation
op()
Return this unit of computation as a single Operation.
org.tensorflow.Operand
abstract Output<T>
asOutput()
Returns the symbolic handle of the tensor.
abstract T
asTensor()
Returns the tensor at this operand.
abstract Shape
shape()
Returns the (possibly partially known) shape of the tensor referred to by the Output of this operand.
abstract Class<T>
type()
Returns the tensor type of this operand
org.tensorflow.ndarray.Shaped
abstract int
rank()
abstract Shape
shape()
abstract long
size()
Computes and returns the total size of this container, in number of values.

Constants

public static final String OP_NAME

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

Constant Value: "SparseMatrixMatMul"

Public Methods

public static SparseMatrixMatMul.Options adjointA (Boolean adjointA)

Parameters
adjointA Indicates whether `a` should be conjugate-transposed.

public static SparseMatrixMatMul.Options adjointB (Boolean adjointB)

Parameters
adjointB Indicates whether `b` should be conjugate-transposed.

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 SparseMatrixMatMul.Options conjugateOutput (Boolean conjugateOutput)

Parameters
conjugateOutput Conjugates the product of `a` and `b`.

public static SparseMatrixMatMul<T> create (Scope scope, Operand<?> a, Operand<T> b, Options... options)

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

Parameters
scope current scope
a A CSRSparseMatrix.
b A dense tensor.
options carries optional attributes values
Returns
  • a new instance of SparseMatrixMatMul

public Output<T> output ()

A dense output tensor.

public static SparseMatrixMatMul.Options transposeA (Boolean transposeA)

Parameters
transposeA Indicates whether `a` should be transposed.

public static SparseMatrixMatMul.Options transposeB (Boolean transposeB)

Parameters
transposeB Indicates whether `b` should be transposed.

public static SparseMatrixMatMul.Options transposeOutput (Boolean transposeOutput)

Parameters
transposeOutput Transposes the product of `a` and `b`.