document / add usage on linear array

Hello.

I have a local array with no format in the repo https://github.com/patrickloeber/pytorch-chatbot/blob/master/train.py#L88 I would like to use LM_Memorization to this llm.

It would be a good use case to start using the software.

Could I do PR please?

Once again, congratulation for your job! It is just amazing.

Best regards.