Hi, thank you for your fast reply.
I want to save output feature of my network. key is the label and value is the feature
k[label_i] = feature(x_i)

I will progressively update the dictionary and use it to compute a cosine similarity with another branch of my network.  Can I use nn.Parameter, nn.Embedding or a simple python dict is okay?
Thank you