Get the “diagonal” entry of a fourth order tensor in python

Question

I have a fourth-order tensor A of shape a * a * a * a. Given a index (i,j,k,l), I want to get all the entry in A with index (i+c, j+d, k+c, l+d), where c and d can be both posit