Обрезать норму градиента в libtorch-C ++ - PullRequest
1 голос
/ 11 марта 2019

Есть ли API для обрезания градиентов сети? Или мне нужно самому развиваться?

Лучший, Afshin

1 Ответ

0 голосов
/ 25 марта 2019

Я нашел один: Tensorat::clamp(const Tensor &self, c10::optional<Scalar> min = c10::nullopt, c10::optional<Scalar> max = c10::nullopt), хотя я тоже реализовал себя.

...