Я планирую реализовать пользовательскую функцию.
Однако если входной тензор является вектором, а выходной тензор - вектором.Результатом будет матрица Якобиана.
Например, функция для вычисления пересечения лучевой сферы.Выходная позиция.Мне нужно сделать это с CUDA.Однако мне может понадобиться сохранить матрицу Якоби во время прямого прохода.Есть ли более эффективный способ сделать это?