


V = torch.randn(nse, dims[2], dims[3])
      >>> S = torch.sparse_coo_tensor(I, V, dims).coalesce()
      >>> D = torch.randn(dims)
      >>> D.sparse_mask(S)
      tensor(indices=tensor([[0, 0, 0, 2],
                             [0, 1, 4, 3]]),
             values=tensor([[[ 1.6550,  0.2397],
                             [-0.1611, -0.0779]],



                        [[ 0.2326, -1.0558],
                         [ 1.4711,  1.9678]],

                        [[-0.5138, -0.0411],
                         [ 1.9417,  0.5158]],

                        [[ 0.0793,  0.0036],
                         [-0.2569, -0.1055]]]),
         size=(5, 5, 2, 2), nnz=4, layout=torch.sparse_coo)
