The easiest way is adding a mask constant.
Like
y = c@(b@(a@x))

You can add a constant like mask = torch.ones((2,2))
ans set mask[0,1]=0
and you can switch your formula to
y = mask.dot(c) @(b@(a@x))

so the gradient of y will not back propagate to c[0,1]