I'm so sorry you and your baby are going through this. No experience with this myself but every dog I've seen with an amputation didn't seem to be a ware that they were missing a leg, got around as well as a dog with four. I did find this article that looked like it had a lot of good information on aftercare. Good luck to you both.
How to Care for a Dog After Amputation Surgery