I hope to rewrite a step function according to my own need.
In this function, I need to new some tensors which are the same size to all weights’ tensors(exclude biases’ tensors).
I hope to get the size from elements in param_groups.But I can’t realize this dict’s elements’ meaning.
Someone can explain this?or where can I learn the information about this?