Badewanne funktioniert nur, wenn du den kleinen festhältst und ihn dann ermuntern kannst zu rudern - also auf der Stelle schwimmen. Hat dann den ähnlichen Effekt wie auf dem Unterwasserlaufband.
So ein Laufband selber zu bauen dürfte technisch gar nicht so einfach sein, denn das Laufband und die gesamte Laufbandführung muss wasserdicht und wasserfest sein. Dazu braucht man schon spezielle. Dazu muss die Wanne entsprechend konstruiert werden und dann ist die Frage, soll es mit Selbstantrieb sein oder soll es mit einem Motor erleichtert werden. Selbstantrieb könnte man vielleicht noch realisieren.
Vielleicht bietet ja irgendein Schwimmbad oder Hundepension oder ähnliches ein Hundeschwimmen in einer Halle an. Das wäre noch eine sehr gute Alternative.
Ansonsten kann ich nur empfehlen, dir bei einer Physiotherapeutin die Möglichkeiten des passiven Muskelaufbaus zeigen zu lassen. Es gibt zwar auch Literatur dazu, aber man sollte es sich schon zeigen lassen.