Tensorflow SparseTensor in 3D - Generating indices

Question

I want to populate the upper triangle of a 3D tf.SparseTensor. To get the correct indices I use triu_indices

triu_indices(3, 1)
<