feature_indices: A `Tensor` of type `int64`. A 2-D int64 tensor of dense_shape [N

Input: ndims].

Output:

Prompt:

 feature_indices: A `Tensor` of type `int64`. A 2-D int64 tensor of dense_shape [N