Function torch::nn::utils::_check_param_device

• Defined in File convert_parameters.h

Function Documentation

inline std::optional<int64_t> torch::nn::utils::_check_param_device(const torch::Tensor &param, std::optional<int64_t> old_param_device)