Generalize `TensorBoardCallback`

[torfjelde]

Currently there's nothing stopping TensorBoardCallback from taking some other logger than TBLogger. The requirements are simply that logging statements are supported for certain types provided in OnlineStats, e.g. Mean and KHist.

Therefore, in the future it might be worth it to add another type parameter for the logger.