Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Due to the broadcasting of 4 arrays (with each of shape Ng x Ng x Ng x Na after broadcasting), too much memory is needed. Therefore, compute Q using a for loop instead of broadcasting and `numpy.sum`. This is a slightly slower computation method, but should take care of the memory issue.
- Loading branch information