Wolfram Language Paclet Repository
Community-contributed installable additions to the Wolfram Language
Tensor Network
returns the tensor network of a circuit | |
TensorNetworkIndexGraph | transforms a tensor network into a new graph with indices as vertices |
TensorNetworkFreeIndices | returns free indices in a tensor network index graph |
ContractTensorNetwork | contracts indices in a tensor network |
Create a quantum circuit:
In[85]:=
circuit=
[{"S","H"2,"X"3,"CNOT","SWAP"{2,3},{1},{2},{3}}];circuit["Diagram"]
 | QuantumCircuitOperator |
Out[86]=
Note in Wolfram quantum framework, each measurement result is saved in an ancillary quantum system (call it a detector, for example). So the true diagram will be something like this:
In[87]:=
circuit["Diagram","ShowExtraQudits"True]
Out[87]=
Note the first measurement is saved in the wire labeled as 0 and the rest goes backward, with negative labels.
Return the tensor network representation of the circuit:
In[88]:=
net=circuit["TensorNetwork",GraphLayout{"LayeredDigraphEmbedding","Orientation"Left}]
Out[88]=
In[90]:=
TensorNetworkContractPath[net,TensorNetworkContractionPath[net]]
Out[90]=
SparseArray
 |  |
|
The tensor network is a graph annotated with tensors and contraction indices.
In[91]:=
GraphQ[net]&&TensorNetworkQ[net]
Out[91]=
True
Lists all annotation keys available for the tensor network:
In[92]:=
AnnotationKeys[{net,0}]
Out[92]=
{Index,Tensor,VertexCoordinates,VertexLabels,VertexShape,VertexShapeFunction,VertexSize,VertexStyle}
Let’s look at tensor, vertex labels, and corresponding indexes in the tensor network:
In[93]:=
list=Developer`FromPackedArray@VertexList[net];TableFormTranspose[Prepend[AnnotationValue[{net,list},#]&/@{"Tensor","Index"},Sort@AnnotationValue[net,VertexLabels]]],
Out[94]//TableForm=
VertexLabel | Tensor | Index | |||||
-20 | SparseArray
| { 3 -2 | |||||
-10 | SparseArray
| { 2 -1 | |||||
00 | SparseArray
| { 1 0 | |||||
1S | SparseArray
| { 1 1 1 1 | |||||
2H | SparseArray
| { 2 2 2 2 | |||||
3X | SparseArray
| 3 3 3 3 | |||||
4CNOT | SparseArray
| { 1 4 2 4 4 1 4 2 | |||||
5SWAP | SparseArray
| 2 5 3 5 5 2 5 3 | |||||
6None | SparseArray
| { 0 6 1 6 6 1 | |||||
7None | SparseArray
| { -1 7 2 7 7 2 | |||||
8None | SparseArray
| -2 8 3 8 8 3 |
Note tensors in the tensor network are mixed type, meaning they consist of so-called “contravariant” (upper) indices and “covariant” (lower) indices. For example, the 2nd measurement (7th vertex), acts on qubit-2 (denoted by contravariant and covariant indices 2) and its result is saved on wire, denoted by the index “-1”.
Vertices correspond to circuit’s operators/gate indices, in addition to “Initial” tensor with index 0 (for the initial state):
In[99]:=
VertexList[net]
Out[99]=
{-2,-1,0,1,2,3,4,5,6,7,8}
In[100]:=
Length@circuit["Flatten"]["Operators"]
Out[100]=
8
In[101]:=
%Max[%%]
Out[101]=
True
Note that each edge represents (i.e., is tagged by) a contraction:
In[102]:=
EdgeList[net]
Out[102]=
01,-12,-23,14,24,45,35,46,57,58
,
1
0
1
1
,
2
-1
2
2
,
3
-2
3
3
,
1
1
4
1
,
2
2
4
2
,
2
4
5
2
,
3
3
5
3
,
1
4
6
1
,
2
5
7
2
,
3
5
8
3
Perform the contraction:
In[103]:=
finalTensor=ContractTensorNetwork[net]
Out[103]=
SparseArray
| |
|
Confirm that the result is the same as default circuit application:
In[104]:=
circuit[]["Tensor"]finalTensor
Out[104]=
True
Another tensor network representation uses indices as graph vertices with tensors as cliques:
In[105]:=
indexNet=TensorNetworkIndexGraph[net,GraphLayout{"LayeredDigraphEmbedding","Orientation"Left}]
Out[105]=
Note in above graph, the directed edges imply tensor contraction; also tensors are cliques in above graph
Free indices are the ones that left after contraction:
Free indices can be extracted as vertices with zero in- and out- degree:
Contraction and Einstein Summation
Contraction and Einstein Summation
What ContractTensorNetwork does is in fact EinsteinSummation.
Show ContractTensorNetwork is the same as EinsteinSummation:
One can compare the performance, on how the relevant computation is done
Perform the contraction in the order of network’s EdgeList:
Optimize the order for contraction, using EinsteinSummation and symbolic tensors package:
Initial state different from ground state
Initial state different from ground state
Note that the initial tensor in the tensor network we studied here was a registered state. Additionally, one can start from any initial state.
Generate a random state:
Initialize the tensor network from above state:
See supplement info, for package-scoped symbols.
Show the tensor contraction is the same as transformation of state by the circuit:
Supplement info
Package-scoped symbols
Package-scoped symbols
FromTensorNetwork
FromTensorNetwork
Any directed graph can be turned into a tensor network, even if the graph is not annotated.