(vars)
| 117 | |
| 118 | @make_recursive_func |
| 119 | def tensor2float(vars): |
| 120 | if isinstance(vars, float): |
| 121 | return vars |
| 122 | elif isinstance(vars, torch.Tensor): |
| 123 | return vars.data.item() |
| 124 | else: |
| 125 | raise NotImplementedError("invalid input type {} for tensor2float".format(type(vars))) |
| 126 | |
| 127 | |
| 128 | def reduce_scalar_outputs(scalar_outputs): |
no outgoing calls
no test coverage detected