Thanks Kushaj for ur reply,
I want to find the mean of angles of different classes. To make it simple, assume that we have angles of different classes stored in in 3D tensors I.e. (10, 2,128), (13,2, 128) …etc. where 1st dimension represents the number of certain classes. 2, 128 is the dimension of angles. Now I want to iterate over all these angles to find the mean of angles.
Cheers,
Angelina