Ejemplos de csr2csr_compress en Python

Ejemplo n.º 1

Archivo: csr.py Proyecto: Hulalazz/cupy

 def eliminate_zeros(self):
     """Removes zero entories in place."""
     if cupy.cuda.runtime.runtimeGetVersion() >= 8000:
         compress = cusparse.csr2csr_compress(self, 0)
     else:
         coo = self.tocoo()
         coo.eliminate_zeros()
         compress = coo.tocsr()
     self.data = compress.data
     self.indices = compress.indices
     self.indptr = compress.indptr

Ejemplo n.º 2

 def eliminate_zeros(self):
     """Removes zero entories in place."""
     if cupy.cuda.runtime.runtimeGetVersion() >= 8000:
         compress = cusparse.csr2csr_compress(self, 0)
     else:
         coo = self.tocoo()
         coo.eliminate_zeros()
         # Because tocsr sums duplicated entries, it cannot keep nnz
         compress = coo.tocsr()
     self.data = compress.data
     self.indices = compress.indices
     self.indptr = compress.indptr

Ejemplo n.º 3

Archivo: csr.py Proyecto: mnicely/cupy

 def eliminate_zeros(self):
     """Removes zero entories in place."""
     compress = cusparse.csr2csr_compress(self, 0)
     self.data = compress.data
     self.indices = compress.indices
     self.indptr = compress.indptr