Package org.tensorflow.op.sparse

Class SparseReshape

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.sparse.SparseReshape

All Implemented Interfaces:

Op

@Operator(group="sparse")
public final class SparseReshape
extends PrimitiveOp

Reshapes a SparseTensor to represent values in a new dense shape.

This operation has the same semantics as reshape on the represented dense tensor. The `input_indices` are recomputed based on the requested `new_shape`.

If one component of `new_shape` is the special value -1, the size of that dimension is computed so that the total dense size remains constant. At most one component of `new_shape` can be -1. The number of dense elements implied by `new_shape` must be the same as the number of dense elements originally implied by `input_shape`.

Reshaping does not affect the order of values in the SparseTensor.

If the input tensor has rank `R_in` and `N` non-empty values, and `new_shape` has length `R_out`, then `input_indices` has shape `[N, R_in]`, `input_shape` has length `R_in`, `output_indices` has shape `[N, R_out]`, and `output_shape` has length `R_out`.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method	Description
static SparseReshape	create(Scope scope, Operand<java.lang.Long> inputIndices, Operand<java.lang.Long> inputShape, Operand<java.lang.Long> newShape)	Factory method to create a class to wrap a new SparseReshape operation to the graph.
Output<java.lang.Long>	outputIndices()	2-D.
Output<java.lang.Long>	outputShape()	1-D.

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 SparseReshape create(Scope scope,
                                   Operand<java.lang.Long> inputIndices,
                                   Operand<java.lang.Long> inputShape,
                                   Operand<java.lang.Long> newShape)

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

Parameters:

scope - current graph scope

inputIndices - 2-D. `N x R_in` matrix with the indices of non-empty values in a SparseTensor.

inputShape - 1-D. `R_in` vector with the input SparseTensor's dense shape.

newShape - 1-D. `R_out` vector with the requested new dense shape.

Returns:

a new instance of SparseReshape

outputIndices

public Output<java.lang.Long> outputIndices()

2-D. `N x R_out` matrix with the updated indices of non-empty values in the output SparseTensor.

outputShape

public Output<java.lang.Long> outputShape()

1-D. `R_out` vector with the full dense shape of the output SparseTensor. This is the same as `new_shape` but with any -1 dimensions filled in.