Returns the device the sparse matrix is on. Returns ------- torch.device The device the sparse matrix is on
(self)
| 60 | |
| 61 | @property |
| 62 | def device(self) -> torch.device: |
| 63 | """Returns the device the sparse matrix is on. |
| 64 | |
| 65 | Returns |
| 66 | ------- |
| 67 | torch.device |
| 68 | The device the sparse matrix is on |
| 69 | """ |
| 70 | return self.c_sparse_matrix.device() |
| 71 | |
| 72 | @property |
| 73 | def row(self) -> torch.Tensor: |