This commit updates the tensor-info.py script to support the option to print the first N values of a tensor when displaying its information. The motivation for this is that it can be useful to inspect some actual values in addition to the shapes of the tensors. |
||
|---|---|---|
| .. | ||
| causal | ||
| embedding | ||
| utils | ||