Försvarsmakten har ett rehabiliteringsansvar för totalförsvarspliktiga som skadas under grund- eller repetitionsutbildning .