dgl.sparse.DiagMatrix.device

property DiagMatrix.device

Returns the device the diagonal matrix is on.

Returns

The device the diagonal matrix is on

Return type

torch.device