Исправление dikiy, 24.02.17 18:38 (текущая версия) :
f должен вернуть тензор размерности i_max на j_max на k_max
А у тебя матрица возвращается 20x20.
Исходная версия dikiy, 24.02.17 18:37: