/print_tensor

Unpack tensors inside list, tuple, dict, generator, and print their shapes

Primary LanguagePython

Stargazers