org.tensorflow.op.sparse

Class SparseFillEmptyRows<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.sparse.SparseFillEmptyRows<T>

Type Parameters:

T - data type for outputValues() output

All Implemented Interfaces:

Op

@Operator(group="sparse")
public final class SparseFillEmptyRows<T>
extends PrimitiveOp

Fills empty rows in the input 2-D `SparseTensor` with a default value.

The input `SparseTensor` is represented via the tuple of inputs (`indices`, `values`, `dense_shape`). The output `SparseTensor` has the same `dense_shape` but with indices `output_indices` and values `output_values`.

This op inserts a single entry for every row that doesn't have any values. The index is created as `[row, 0, ..., 0]` and the inserted value is `default_value`.

For example, suppose `sp_input` has shape `[5, 6]` and non-empty values:

[0, 1]: a [0, 3]: b [2, 0]: c [3, 1]: d

Rows 1 and 4 are empty, so the output will be of shape `[5, 6]` with values:

[0, 1]: a [0, 3]: b [1, 0]: default_value [2, 0]: c [3, 1]: d [4, 0]: default_value

The output `SparseTensor` will be in row-major order and will have the same shape as the input.

This op also returns an indicator vector shaped `[dense_shape[0]]` such that

empty_row_indicator[i] = True iff row i was an empty row.

And a reverse index map vector shaped `[indices.shape[0]]` that is used during backpropagation,

reverse_index_map[j] = out_j s.t. indices[j, :] == output_indices[out_j, :]

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static <T> SparseFillEmptyRows<T>	create(Scope scope, Operand<Long> indices, Operand<T> values, Operand<Long> denseShape, Operand<T> defaultValue) Factory method to create a class wrapping a new SparseFillEmptyRows operation.
Output<Boolean>	emptyRowIndicator() 1-D.
Output<Long>	outputIndices()
Output<T>	outputValues() 1-D.
Output<Long>	reverseIndexMap() 1-D.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

public static <T> SparseFillEmptyRows<T> create(Scope scope,
                                                Operand<Long> indices,
                                                Operand<T> values,
                                                Operand<Long> denseShape,
                                                Operand<T> defaultValue)

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

Parameters:

scope - current scope

indices - 2-D. the indices of the sparse tensor.

values - 1-D. the values of the sparse tensor.

denseShape - 1-D. the shape of the sparse tensor.

defaultValue - 0-D. default value to insert into location `[row, 0, ..., 0]` for rows missing from the input sparse tensor. output indices: 2-D. the indices of the filled sparse tensor.

Returns:

a new instance of SparseFillEmptyRows

outputIndices

public Output<Long> outputIndices()

outputValues

public Output<T> outputValues()

1-D. the values of the filled sparse tensor.

emptyRowIndicator

public Output<Boolean> emptyRowIndicator()

1-D. whether the dense row was missing in the input sparse tensor.

reverseIndexMap

public Output<Long> reverseIndexMap()

1-D. a map from the input indices to the output indices.