Tyler Yep

What is this feature supposed to do exactly? I tried it and it transformed the following string, which seems unideal to me, since it breaks the max line length of...

Are you dedicated to supporting Python 2.6 at this point? All versions of up to Python 3.5 have reached end-of-life by now (https://endoflife.date/python), so it seems wise to move along...

Resubmitted the PR. I can go ahead and create the PR for the black formatting, but could you fix the travis CI integration first? I believe you just need to:...

From the README: `python -m pytest` I tried it just now and it works.

Torchinfo currently does not support counting MACs of these raw operations. The current implementation has a lot of room for improvement: https://github.com/TylerYep/torchinfo/blob/main/torchinfo/layer_info.py#L188 PRs addng or improving this functionality are welcome!

Thanks for reporting this issue. Any PR or help fixing this is much appreciated!

This is happening because torchinfo currently does not have a sophisticated way of handling dicts containing tensors. Would appreciate PRs improving this functionality! The `calculate_size` function is where the changes...

Good observation. I would appreciate any PRs addressing this issue!

Thanks for reporting this issue. numpy arrays and size functionality is not supported in torchinfo at the moment. PRs to add support for this output type are much appreciated!

This is a feature request, in #55 I mentioned there is only preliminary support for JIT (as in no crashes). I am unfamiliar with the mechanics of JIT, so maybe...