提取tf张量的特定子指标

Question

我有一个t[i,j,k,...]的张量len(shape) = n，我需要得到t[i,i,k,...]的{​​{1}}。 Tensor的形状可能随时间而变化，但总是len(shape) = n-1。 shape[1] == shape[0]没有解决问题，因为除了前两个之外的所有其他索引都是任意的。我已经搜索了很长一段时间，但仍无法找到解决方案。拜托，帮助我！