Use str(v) instead of '%d' % v to stringify sizes

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs

MIT License

3.24k stars 279 forks source link

Use str(v) instead of '%d' % v to stringify sizes #81

Open davidberard98 opened 10 months ago

davidberard98 commented 10 months ago

Nested tensors can have sizes that cannot be converted to ints (e.g. singleton symints representing the jagged dimension, like [4, j0, 8]). str() is supported on singleton symints.

albanD commented 9 months ago

@szagoruyko could you merge this? I'm happy to handle it as well if you want to give me write access.