Function dense_mask

tests/python/pytorch/sparse/utils.py:158–163

(dense, sparse)

def dense_mask(dense, sparse):
    ret = torch.zeros_like(dense)
    row, col = sparse.coo()
    for r, c in zip(row, col):
        ret[r, c] = dense[r, c]
    return ret