org.tensorflow.op.linalg.sparse

Class SparseMatrixSoftmax

java.lang.Object

org.tensorflow.op.RawOp

org.tensorflow.op.linalg.sparse.SparseMatrixSoftmax

All Implemented Interfaces:

Shaped, Op, Operand<TType>

public final class SparseMatrixSoftmax
extends RawOp
implements Operand<TType>

Calculates the softmax of a CSRSparseMatrix.

Calculate the softmax of the innermost dimensions of a SparseMatrix.

Missing values are treated as `-inf` (i.e., logits of zero probability); and the output has the same sparsity structure as the input (though missing values in the output may now be treated as having probability zero).

Field Summary

Fields

Modifier and Type	Field and Description
static String	OP_NAME The name of this op, as known by TensorFlow core engine

Fields inherited from class org.tensorflow.op.RawOp

operation

Method Summary

Modifier and Type	Method and Description
Output<TType>	asOutput() Returns the symbolic handle of the tensor.
static <T extends TNumber> SparseMatrixSoftmax	create(Scope scope, Operand<?> logits, Class<T> type) Factory method to create a class wrapping a new SparseMatrixSoftmax operation.
Output<?>	softmax() A CSRSparseMatrix.

Methods inherited from class org.tensorflow.op.RawOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.tensorflow.Operand

asTensor, shape, type

Methods inherited from interface org.tensorflow.op.Op

env, op

Methods inherited from interface org.tensorflow.ndarray.Shaped

rank, size

Field Detail

OP_NAME

public static final String OP_NAME

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

See Also:

Constant Field Values

Method Detail

create

@Endpoint(describeByClass=true)
public static <T extends TNumber> SparseMatrixSoftmax create(Scope scope,
                                                                                             Operand<?> logits,
                                                                                             Class<T> type)

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

Parameters:

scope - current scope

logits - A CSRSparseMatrix.

type -

Returns:

a new instance of SparseMatrixSoftmax

softmax

public Output<?> softmax()

A CSRSparseMatrix.

asOutput

public Output<TType> asOutput()

Description copied from interface: Operand

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.

Specified by:

asOutput in interface Operand<TType>

See Also:

OperationBuilder.addInput(Output)