Skip to content

lwelzel/td-viz

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

td-viz

Visualization of Tensor Decompositions

Produce publication quality tikz figures of tensor decompositions for use in Latex.

3D visualizations of tensor decompositions

Define the parameters of the decomposition and automatically produce the appropriate visualization. Currently only supports Tensor Train and Tensor Ring decompositions.

Order-4 Tensor Train

image

Order-4 Tensor Ring

image

2D visualizations of tensor decompositions in tensor network diagrams

Tensor network diagrams. Currently does not support automatic figure generation. Requires the tikz-network package.

Tensor

image

CP Decomposition

image

Tucker Decomposition

image

Tensor Train Decomposition

image

Tensor Ring Decomposition

In either TT or TR formatting: image

Usage

Please attribute me (Lukas Welzel) if you use this package.

Geometric tikz figures

Use the provided python script to produce either TT or TR decomposition figures.

TND style figures

Use the provided Latex baselines.

Requirements

Geometric tikz figures

  • Python 3.6+
  • numpy

TND style figures

  • tikz-network

Existing Tools:

Motivation

When working on TD I found that there were no good tools to produce high quality plots of TD in either TND or, even less so, in their geometric form. This package aims to provide a simple way to produce high quality figures of tensor decompositions. Currently, it is a bit of a hack job but if you need to produce figures or equations that look somewhat like the cases that are shown above it will do the job. In the future, I aim to clean this up and make it a bit more user-friendly. I do not have a very strong background in TD or TND so please feel free to contribute to this project or notify me of any errors.