(obj)
| 16 | return Variable(obj) |
| 17 | |
| 18 | def as_numpy(obj): |
| 19 | if isinstance(obj, collections.Sequence): |
| 20 | return [as_numpy(v) for v in obj] |
| 21 | elif isinstance(obj, collections.Mapping): |
| 22 | return {k: as_numpy(v) for k, v in obj.items()} |
| 23 | elif isinstance(obj, Variable): |
| 24 | return obj.data.cpu().numpy() |
| 25 | elif torch.is_tensor(obj): |
| 26 | return obj.cpu().numpy() |
| 27 | else: |
| 28 | return np.array(obj) |
| 29 | |
| 30 | def mark_volatile(obj): |
| 31 | if torch.is_tensor(obj): |