예제 #1
0
파일: Tensor.py 프로젝트: xichuang/HiParTI
 def mulMatrix(self, tensor, mode, type="default", testing=False):
     result = sCOOTensor()
     if type == "default":
         serial = 0  #TODO make params
     if type == "serial":
         PTI.ptiSemiSparseTensorMulMatrix(result.address, self.address,
                                          tensor.address, mode)
     elif type == "GPU":
         PTI.ptiCudaSemiSparseTensorMulMatrix(result.address, self.address,
                                              tensor.address, mode)
     if not testing: return result
     else: PTI.ptiFreeSemiSparseTensor(result.address)
예제 #2
0
파일: Tensor.py 프로젝트: xichuang/HiParTI
 def free(self):
     PTI.ptiFreeSemiSparseTensor(self.address)