`torch.device`: The device on which the module is (assuming that all the module parameters are on the same device).
(self)
| 1004 | |
| 1005 | @property |
| 1006 | def device(self) -> torch.device: |
| 1007 | """ |
| 1008 | `torch.device`: The device on which the module is (assuming that all the module parameters are on the same |
| 1009 | device). |
| 1010 | """ |
| 1011 | return get_parameter_device(self) |
| 1012 | |
| 1013 | @property |
| 1014 | def dtype(self) -> torch.dtype: |