Marcel-Rodekamp / NSL

Nanosystem Simulation Library (NSL) implements statistical simulations for systems on the nanoscale
Other
2 stars 0 forks source link

NSL::Tensor.mat_mul reduction #66

Open evanberkowitz opened 2 years ago

evanberkowitz commented 2 years ago

In the fermion matrix we can easily construct a stack of matrices of shape (nt, nx, nx) and we need to take the matrix product of all the slices one after the next.

In #65 this reduction is implemented manually via a for loop, but it would be nice to provide this reduction as part of NSL::Tensor.

evanberkowitz commented 1 year ago

I'm asking for expert help in the torch forums.