tft.experimental.annotate_sparse_output_shape

Annotates a sparse output to have a given dense_shape.

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