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)
def free(self): PTI.ptiFreeSemiSparseTensor(self.address)