# Importing the library
import tensorflow as tf
  
# Initializing the input
data = tf.constant([1, 2, 3])
  
# Printing the input
print('data: ', data)
  
# Calculating result
res = tf.IndexedSlices(data, [0])
  
# Finding Graph 
indices = res.indices
  
# Printing the result
print('Indices: ', indices)

data:  tf.Tensor([1 2 3], shape=(3, ), dtype=int32)
Indices:  [0]

# Importing the library
import tensorflow as tf
  
# Initializing the input
data = tf.constant([[1, 2, 3], [4, 5, 6]])
  
# Printing the input
print('data: ', data)
  
# Calculating result
res = tf.IndexedSlices(data, [0, 1])
  
# Finding Graph 
indices = res.indices
  
# Printing the result
print('Indices: ', indices)

data:  tf.Tensor(
[[1 2 3]
 [4 5 6]], shape=(2, 3), dtype=int32)
Indices:  [0, 1]