BasicSelfKernel in module.py

[jxtxwsbn]

Hi,

After several tests, I think the line 183 of module.py (x@kernel) has a problem.
It calculates torch.matmul(x,kernel) instead of torch.einsum('ij,pj ->pi',kernel,x).