tfm.vision.sampling_ops.indices_to_dense_vector

Creates dense vector with indices set to specific value and rest to zeros.

tfm.vision.sampling_ops.indices_to_dense_vector(
    indices, size, indices_value=1.0, default_value=0, dtype=tf.float32
)

This function exists because it is unclear if it is safe to use tf.sparse_to_dense(indices, [size], 1, validate_indices=False) with indices which are not ordered. This function accepts a dynamic size (e.g. tf.shape(tensor)[0])

indices 1d Tensor with integer indices which are to be set to indices_values.
size scalar with size (integer) of output Tensor.
indices_value values of elements specified by indices in the output vector
default_value values of other elements in the output vector.
dtype data type.

dense 1D Tensor of shape [size] with indices set to indices_values and the rest set to default_value.