Тыкнули меня лицом в pytorch
Вот торчу теперь и не втыкаю, не чистая это дрянь
Вот так сразу не понять где данные клонируются а где всё ссылается на одну область в памяти.

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