Package org.tensorflow.op.sparse

Class SparseTensorDenseMatMul<U>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.sparse.SparseTensorDenseMatMul<U>

Type Parameters:

U - data type for product() output

All Implemented Interfaces:

Op, Operand<U>

@Operator(group="sparse")
public final class SparseTensorDenseMatMul<U>
extends PrimitiveOp
implements Operand<U>

Multiply SparseTensor (of rank 2) "A" by dense matrix "B".

No validity checking is performed on the indices of A. However, the following input format is recommended for optimal behavior:

if adjoint_a == false: A should be sorted in lexicographically increasing order. Use SparseReorder if you're not sure. if adjoint_a == true: A should be sorted in order of increasing dimension 1 (i.e., "column major" order instead of "row major" order).

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	SparseTensorDenseMatMul.Options	Optional attributes for SparseTensorDenseMatMul

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method	Description
static SparseTensorDenseMatMul.Options	adjointA(java.lang.Boolean adjointA)
static SparseTensorDenseMatMul.Options	adjointB(java.lang.Boolean adjointB)
Output<U>	asOutput()	Returns the symbolic handle of a tensor.
static <U,T extends java.lang.Number> SparseTensorDenseMatMul<U>	create(Scope scope, Operand<T> aIndices, Operand<U> aValues, Operand<java.lang.Long> aShape, Operand<U> b, SparseTensorDenseMatMul.Options... options)	Factory method to create a class to wrap a new SparseTensorDenseMatMul operation to the graph.
Output<U>	product()

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Method Detail

create

public static <U,T extends java.lang.Number> SparseTensorDenseMatMul<U> create(Scope scope,
                                                                               Operand<T> aIndices,
                                                                               Operand<U> aValues,
                                                                               Operand<java.lang.Long> aShape,
                                                                               Operand<U> b,
                                                                               SparseTensorDenseMatMul.Options... options)

Factory method to create a class to wrap a new SparseTensorDenseMatMul operation to the graph.

Parameters:

scope - current graph scope

aIndices - 2-D. The `indices` of the `SparseTensor`, size `[nnz, 2]` Matrix.

aValues - 1-D. The `values` of the `SparseTensor`, size `[nnz]` Vector.

aShape - 1-D. The `shape` of the `SparseTensor`, size `[2]` Vector.

b - 2-D. A dense Matrix.

options - carries optional attributes values

Returns:

a new instance of SparseTensorDenseMatMul

adjointA

public static SparseTensorDenseMatMul.Options adjointA(java.lang.Boolean adjointA)

Parameters:

adjointA - Use the adjoint of A in the matrix multiply. If A is complex, this is transpose(conj(A)). Otherwise it's transpose(A).

adjointB

public static SparseTensorDenseMatMul.Options adjointB(java.lang.Boolean adjointB)

Parameters:

adjointB - Use the adjoint of B in the matrix multiply. If B is complex, this is transpose(conj(B)). Otherwise it's transpose(B).

product

public Output<U> product()

asOutput

public Output<U> asOutput()

Description copied from interface: Operand

Returns the symbolic handle of a 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.

Specified by:

asOutput in interface Operand<U>

See Also:

OperationBuilder.addInput(Output)