Hello community,
I’m trying to optimize some PyTorch code and I’m interested in comparing the computational asymptotic complexity of different PyTorch functions. For example, I would like to compare the complexity of torch.cat vs torch.stack. I am talking about the most frequent functions that are present in let’s say 90% of torch use cases. Does anyone know of a chart or resource that already provides this information?
Alternatively, I would be interested in creating such a chart myself. Does anyone know of a platform where I could upload the chart so that others could access it and potentially update it in the future if there are changes to the PyTorch codebase?
Thank you in advance for any help or suggestions!