Distinction between agent i and j not clear

[sapan-ostic]

Hi,

The paper mentions agent i (agent whose trajectory is to be predicted ) and other agents j (neighbors). However, I couldn't find such distinction in the code. Can you please explain?

Also, how is the information about the neighbors stored in the dataset_obsv?
dim[dataset_obsv] = NxTx2 (line 89, train.py)
So does N here refers to the sample size?

I am confused because SocialFeatures() function mentions N as well and seems like N = No. of agents in the scene, here.

Thanks.