Torch.utils.tensorboard fails with multiple inputs model

As a side note, I already verified that the tensors are both FloatTensors as per this thread RuntimeError: Integer division of tensors using div or / is no longer supported, and in a future release div will perform true division as in Python 3. Use true_divide or floor_divide (// in Python) instead