tf.raw_ops.SparseReshape
Reshapes a SparseTensor to represent values in a new dense shape.
tf.raw_ops.SparseReshape(
input_indices, input_shape, new_shape, name=None
)
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.
| Args | |
|---|---|
input_indices | A Tensor of type int64. 2-D. N x R_in matrix with the indices of non-empty values in a SparseTensor. |
input_shape | A Tensor of type int64. 1-D. R_in vector with the input SparseTensor's dense shape. |
new_shape | A Tensor of type int64. 1-D. R_out vector with the requested new dense shape. |
name | A name for the operation (optional). |
| Returns | |
|---|---|
A tuple of Tensor objects (output_indices, output_shape). | |
output_indices | A Tensor of type int64. |
output_shape | A Tensor of type int64. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/SparseReshape