tft.experimental.annotate_sparse_output_shape

Annotates a sparse output to have a given dense_shape.

tft.experimental.annotate_sparse_output_shape(
    tensor: tf.SparseTensor, shape: Union[Sequence[int], tf.Tensor]
)

Args
tensor	An SparseTensor to be annotated.
shape	A dense_shape to annotate tensor with. Note that this shape does not include batch_size.