Restore Translation Using Equivariant Neural Networks