def round_loss(tens, dec=0):
    """from an input scalar tensor or int/float returns rounded int/float"""
    if type(tens) is int or type(tens) is float:
        return round(tens, dec)
    else:
        return round(tens.item(), dec)