标签： python tensorflow machine-learning tensor
假设我有一个等级四张量：T_{i,j,p,q},并且我想找到前两个维度的对角线元素以形成一个新的张量：T_{i,i,p,q}到T_{i,p,q}。我应该怎么做张量流？
T_{i,j,p,q},
T_{i,i,p,q}
T_{i,p,q}
由于