Add class to handle tensorboard outputs

Class to easily handle tensorboard outputs.