(args, indent="")
| 313 | |
| 314 | |
| 315 | def print_inputs(args, indent=""): |
| 316 | for t in args: |
| 317 | if isinstance(t, torch.Tensor): |
| 318 | print(indent, t.stride(), t.shape, t.device, t.dtype) |
| 319 | elif isinstance(t, list) or isinstance(t, tuple): |
| 320 | print(indent, type(t)) |
| 321 | print_inputs(list(t), indent=indent + " ") |
| 322 | else: |
| 323 | print(indent, t) |
| 324 | |
| 325 | |
| 326 | def check_wrapper(fn): |