自定义nn.module层的注意事项
自定义nn.module层的注意事项
待办

使用张量构建参数,为什么不用require_grad=true,而是加到nn.Parameter中
要使用优化器优化要使用nn.parameter这个类,所有的参数都在这里边,所以要把参数都加进去
这样自定义的MyLinear和 nn.Linear一模一样
待办
要使用优化器优化要使用nn.parameter这个类,所有的参数都在这里边,所以要把参数都加进去
这样自定义的MyLinear和 nn.Linear一模一样