Fix printing of Tensors

pyro-ppl / funsor

Functional tensors for probabilistic programming

https://funsor.pyro.ai

Apache License 2.0

236 stars 20 forks source link

Fix printing of Tensors #511

Closed fritzo closed 3 years ago

fritzo commented 3 years ago

Addresses ugly printing in #509

I'd like to merge this before #509 so the tutorial looks nice.

Tested

[x] unit test for domains
[x] unit test for tensors

ordabayevy commented 3 years ago

Somewhat related to this, when Tensors are printed the first row is misaligned because of the additional Tensor( string (see output of cell 6 in #509). Do you think that should also be fixed @fritzo?

fritzo commented 3 years ago

Tensors are printed the first row is misaligned

@ordabayevy Good point, I've fixed this.

fritzo commented 3 years ago

Thanks for reviewing @fehiepsi !