trancelestial:

If you need more specific use case info, I would be happy to provide it.


I would be interested in hearing more about your use case.
If I understand it correctly, you would like to be able to use something like:
params = nn.ParamArray([[param1, param2], [param3, param4]])
p = params[0, 1]