Applies set operation along last dimension of 2 SparseTensor
inputs.
tf.raw_ops.SparseToSparseSetOperation( set1_indices, set1_values, set1_shape, set2_indices, set2_values, set2_shape, set_operation, validate_indices=True, name=None )
See SetOperationOp::SetOperationFromContext for values of set_operation
.
If validate_indices
is True
, SparseToSparseSetOperation
validates the order and range of set1
and set2
indices.
Input set1
is a SparseTensor
represented by set1_indices
, set1_values
, and set1_shape
. For set1
ranked n
, 1st n-1
dimensions must be the same as set2
. Dimension n
contains values in a set, duplicates are allowed but ignored.
Input set2
is a SparseTensor
represented by set2_indices
, set2_values
, and set2_shape
. For set2
ranked n
, 1st n-1
dimensions must be the same as set1
. Dimension n
contains values in a set, duplicates are allowed but ignored.
If validate_indices
is True
, this op validates the order and range of set1
and set2
indices.
Output result
is a SparseTensor
represented by result_indices
, result_values
, and result_shape
. For set1
and set2
ranked n
, this has rank n
and the same 1st n-1
dimensions as set1
and set2
. The nth
dimension contains the result of set_operation
applied to the corresponding [0...n-1]
dimension of set
.
Args | |
---|---|
set1_indices | A Tensor of type int64 . 2D Tensor , indices of a SparseTensor . Must be in row-major order. |
set1_values | A Tensor . Must be one of the following types: int8 , int16 , int32 , int64 , uint8 , uint16 , string . 1D Tensor , values of a SparseTensor . Must be in row-major order. |
set1_shape | A Tensor of type int64 . 1D Tensor , shape of a SparseTensor . set1_shape[0...n-1] must be the same as set2_shape[0...n-1] , set1_shape[n] is the max set size across 0...n-1 dimensions. |
set2_indices | A Tensor of type int64 . 2D Tensor , indices of a SparseTensor . Must be in row-major order. |
set2_values | A Tensor . Must have the same type as set1_values . 1D Tensor , values of a SparseTensor . Must be in row-major order. |
set2_shape | A Tensor of type int64 . 1D Tensor , shape of a SparseTensor . set2_shape[0...n-1] must be the same as set1_shape[0...n-1] , set2_shape[n] is the max set size across 0...n-1 dimensions. |
set_operation | A string . |
validate_indices | An optional bool . Defaults to True . |
name | A name for the operation (optional). |
Returns | |
---|---|
A tuple of Tensor objects (result_indices, result_values, result_shape). | |
result_indices | A Tensor of type int64 . |
result_values | A Tensor . Has the same type as set1_values . |
result_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.3/api_docs/python/tf/raw_ops/SparseToSparseSetOperation