Allow NaN in batch fitting

Make the torch.mean calls NaN compatible such that regularizers can be computed without introducing zero padding

Durchsuche den Batch-Fitting-Code nach torch.mean-Aufrufen und verfolge, wie deren Ergebnisse in die Berechnung der Regularisierer einfließen. Prüfe, wie NaN-Werte und Zero-Padding derzeit behandelt werden, und stelle anschließend sicher, dass Regularisierer berechnet werden können, ohne Zero-Padding einzuführen, wobei das bestehende Fitting-Verhalten erhalten bleibt.

Vom Indexierungsmodell aus dem Issue-Text verfasst.