(tensors)
| 2 | |
| 3 | |
| 4 | def _assert_contiguous(tensors): |
| 5 | data_ptr = None |
| 6 | for t in tensors: |
| 7 | if data_ptr is not None: |
| 8 | assert t.data_ptr() == data_ptr |
| 9 | data_ptr = t.data_ptr() + t.numel() * t.element_size() |
| 10 | |
| 11 | |
| 12 | def flatten_params(param_set, chunk=None): |