Я столкнулся со следующей проблемой:
Предположим, нам дан список матрично-матричных умножений со следующими формами матриц:
[(3x2)*(2x3),
(5x2)*(2x5),
(4x2)*(2x4),
(2x2)*(2x2)]
Как этот список умножения матрицы на матрицу можно выполнить через tf с использованием графических процессоров? Как мы можем сложить массив неправильных форм (потому что, насколько мне известно, формы должны быть равны, чтобы применить их позже, например, tf.map_fn?)?
Имеет ли смысл в tf выполнять многократное умножение матриц на матрицу (эр) с использованием графического процессора? Что в целом является хорошей стратегией / эвристикой для распределения задач, таких как умножение матриц или eigendecomp. на графическом процессоре?
С наилучшими пожеланиями