xl0 / lovely-tensors

Tensors, ready for human consumption

Home Page:https://xl0.github.io/lovely-tensors

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can you fine tune the view in any way?

Meehai opened this issue · comments

For example, is there any way to disable showing if there are NaNs, or to simply show just the shape?

Perhaps set_config() could accept a callback that receives a tensor and outputs a string (and defaults to current implementation), so we can fine tune it further based on custom logic?

Heya. I'm not against it, but this would definitely require a bit of effort because this functionality is split and semi-duplicated between lovely-numpy and tensors/jax/etc. I don't think I'd have time for it, but I would be open to a PR, although, as I said, it's not a trivial change.