Searched defs:SoftshrinkOptions (Results 1 – 2 of 2) sorted by relevance
531 struct TORCH_API SoftshrinkOptions { struct
22 SoftshrinkOptions::SoftshrinkOptions(double lambda) : lambda_(lambda) {} in SoftshrinkOptions() function in torch::nn::SoftshrinkOptions