Any API complexity table?

1

Hi there,

Thanks for the great framework!

Since the runtime may be affected by various factors. Is it possible for users to report and compare the model complexity by summing up the complexity for each operation?

While FLOPs work for traditional layers, the complexity/FLOPs for APIs on sparse tensors, indexing, etc. operations are unclear, and we suppose it should relate to the implementation.

We are wondering if there are any place we can refer to such complexities. Thanks!

2

Unfortunately not, but this is something that we have always thought was a good idea, just haven’t gotten around to implementing. See https://github.com/pytorch/pytorch/issues/5013 (although I see you have seen it already!)

1 Like