这是正确的。在PyTorch中,只有nn.Parameter对象才会被优化器优化并参与反向传播。因此,如果需要让随机参数或随机操作参与训练过程,需要将其封装成nn.Parameter对象。

就是说无论user设定任何的随机参数或random操作只要将其设定为nnparameter就可以参与学习和反向传播。

原文地址: https://www.cveoy.top/t/topic/byc1 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录