Hello there,
I am working on a project where the output of the network is a minibatch of 12 channel tensors of size 500x500. In the loss i want to take a each of these tensors and access each element through it arbirtrarily(I will have the information of which coordinate I will go to). The reason I mention arbitrary is because I cannot use slicing or any such thing to parallelize it (I guess…). Is there any way I can do it? I know if I manually loop through the elements one by one its gonna slow down my training immensely. Any idea what should I look into?
Thanks in advance!