CODEWITHSUNDEEP
arraysfortran
Qwertuy
Qwertuy

Reputation: 198

Efficient way to perform tensor products in Fortran

I need to perform some tensor products and contractions on some large arrays in Fortran. Sometimes they are vectors or matrices and sometimes some of the objects involved are 3-arrays or 4-arrays.

Of course, it is very easy to write a subroutine achieving this with some nested loops, and that's just what I've done. But I have to call this subroutine with all its loops a lot of times for very large arrays, and I was just wondering whether there is some optimized function or subroutine implemented in Fortran which I could benefit from.

Upvotes: 4

Views: 3583

Answers (1)

ftiaronsem
ftiaronsem

Reputation: 1584

Last time I looked (about a year ago) I did not find a high performance general purpose tensor product library in Fortran. I think one of the reason for this might be Fortran's cumbersome way of resizing arrays, which is a constant requirement when dealing with tensors.

If you only need multiplication you might be able to get away with using your own code. However if you need high performance, or more general operations, I would highly recommend writing a C interface and using one of the excellent C++ libraries out there, which are probably already optimized for your type of application:

Physics: http://itensor.org/

Machine learning: https://github.com/tensorflow/tensorflow

These are only examples. For a more complete listing see: Tensor multiplication library

Upvotes: 3

Related Questions