OOM

[CodeSageMaster]

Hello, thanks for your open source contribution, matrix multiplication and cat operations are used when implementing equation 6 in your code, and when executing slightly larger datasets such as PUBMED, AMAC, Corafull, etc., the error of out of memory is prompted, is it possible to improve the implementation method to accommodate large datasets？