How can I visualize Tensor data from the Burn framework?

[callmeitachi]

Hi everyone, I was wondering if the Burn framework provides a quick way to convert Tensor data to ndarray format? I'm currently learning Burn and would like to visualize some data. The plotting libraries I know about are plotters and plotly - does Burn have any built-in visualization libraries for this purpose?

[laggui]

If the tensor data is an image, we just added save_tensor_as_image in burn-vision recently.

Otherwise, there are no built-in way to visualize the tensors other than the Display and Debug implementations. But people have built their own visualization tools in the meantime (e.g., https://github.com/VirtualNonsense/sketchy_pix2pix/ uses rerun to visualize gradients).

What kind of visualization are you looking for?