How to extract t[i,i,n] using tensorflow? : For example:

    t = tf.Constant(np.array([[[1,2],[3,4]],[[5,6],[7,8]]]))
    # I want to extract h = [[1,2],[7,8]]

How to use tensorflow to do this? Thank you!