Hi All,
I have a list of tensors and I have say 500 classes, I want to calculate the mean vector for each class in a differentiable way.
here is my code:
y is the labels
H =torch.randn (800,800)
C = 500
mean_vectors_list = []
mean_vectors_list.append([torch.mean(H[y == cl], dim=0, dtype=torch.float32) for cl in 
range(C)])

The problem is some times H tensor doesn’t have some classes as it’s a randomly selected batch
that gives me Nan.
How to use .scatter_add_ to produce the desired output with cuda