search

parrt / tensor-sensor

The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.
https://github.com/parrt/tensor-sensor
MIT License
794 stars 40 forks source link

Ideas for 3D+ shapes #2

Open parrt opened 4 years ago

parrt commented 4 years ago

https://www.tensorflow.org/guide/tensor

ConstantinosM commented 4 years ago

I have no idea how you produce these beautiful and very useful visualizations, so I do not know if my idea helps you. I was thinking something along these lines:

image

This is for a (4,8,64,64,3) tensor. You might use a cube as a "unit" for higher order tensors. For example for (10,4,8,64,64,3) you build a cube (10,4,8) which is consist of smaller cubes (64,64,3).