Is there a simple way to zero a diagonal of a pytorch tensor?
for example I have:
tensor([[2.7183, 0.4005, 2.7183, 0.5236], [0.4005, 2.7183, 0.4004