Эффективный способ выполнения тензорных продуктов в Фортране

Question

Мне нужно выполнить некоторые тензорные произведения и сокращения на некоторых больших массивах в Фортране. Иногда они являются векторами или матрицами, и иногда некоторые из задействованных объектов представляют собой 3-массивы или 4-массивы.

Конечно, очень легко написать подпрограмму, обеспечивающую это с помощью некоторых вложенных циклов, и это именно то, что я сделал. Но мне приходится многократно вызывать эту подпрограмму со всеми ее циклами для очень больших массивов, и мне просто интересно, есть ли какая-то оптимизированная функция или подпрограмма, реализованная в Fortran, в которой я мог бы воспользоваться.

Answer 1

Последний раз, когда я смотрел (около года назад), я не нашел высокопроизводительную библиотеку продуктов тензора общего назначения в Фортране. Я думаю, что одной из причин этого может быть громоздкий способ изменения размера массивов Fortran, что является постоянным требованием при работе с тензорами.

Если вам нужно только умножение, вы можете уйти с помощью своего собственного кода. Однако, если вам нужна высокая производительность, или более общие операции, я настоятельно рекомендую писать интерфейс C и с помощью одного из отличных C++ библиотек там, которые, вероятно, уже оптимизированные для вашего типа применения:

физика: http://itensor.org/

Машинное обучение: https://github.com/tensorflow/tensorflow

это только примеры. Для получения более полного списка см .: Tensor multiplication library