У меня есть два тензора одинаковой формы: [64,32,512,16,16,3] Я вычитаю их и получаю результат с одинаковой формой.и затем я использую reduce_sum, reduce_min, reduce_max в последних 3 измерениях, чтобы результат имел форму [64,32,512].
Проблема в том, что эти промежуточные тензоры с формой [64,32,512,16,16,3] огромныи поэтому я сталкиваюсь с проблемами памяти.Есть ли способ решить эту проблему?