Hi Thomas,
Thank you for your reply. Unfortunately I have to deal with tensors with large size, roughly around 250000. I have tried broadcasting, it’s slow and the computation of gradient is even slower.
Do you have any idea how to deal with tensor with large size?
Best regards
Peng