标签: libtorch
是否有用于裁剪网络渐变的API?还是我需要自我发展?
最好, 苦参碱
答案 0 :(得分:0)
我发现了一个: Tensorat::clamp(const Tensor &self, c10::optional<Scalar> min = c10::nullopt, c10::optional<Scalar> max = c10::nullopt),尽管我也实现了自己。
Tensorat::clamp(const Tensor &self, c10::optional<Scalar> min = c10::nullopt, c10::optional<Scalar> max = c10::nullopt)