Wolfram Language Paclet Repository
Community-contributed installable additions to the Wolfram Language
Tensor Network
Tensor Network
| | |
QuantumCircuitOperator | 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[1]:=
circuit=
[{"S","H"2,"X"3,"CNOT","SWAP"{2,3},{1},{2},{3}}];circuit["Diagram"]
 | QuantumCircuitOperator |
In the Wolfram Quantum Framework, measurement outcomes are recorded in an ancillary quantum register (serving as a detector), so the complete circuit diagram includes both the system and this detector subsystem.
In[2]:=
circuit["Diagram","ShowExtraQudits"True]
The first measurement result is stored in wire index 0, with subsequent results assigned to decreasing (negative) wire indices.
Returns the tensor network representation corresponding to the given quantum circuit:
In[3]:=
net=circuit["TensorNetwork",GraphLayout{"LayeredDigraphEmbedding","Orientation"Left}]
The tensor network is a graph annotated with tensors and contraction indices.
In[4]:=
GraphQ[net]&&TensorNetworkQ[net]
For a graph to be a tensor network, two main conditions must be met, which is what does. First, every vertex in the graph must have properly formatted indices consisting only of Subscript or Superscript expressions, and each vertex cannot contain any duplicate indices - all indices at any single vertex must be unique from each other.
TensorNetworkQ
Second, the tensor rank compatibility condition must be satisfied. This can happen in two ways: either each vertex's tensor rank must exactly match the number of indices assigned to that vertex, and when counting all index occurrences throughout the entire graph, every distinct index must appear exactly twice across all vertices (ensuring proper pairing for tensor contraction), or alternatively, all tensors in the network must have rank zero representing scalar tensors with no indices.
Lists all annotation keys available for the tensor network:
In[5]:=
AnnotationKeys[{net,0}]
Let’s look at tensor, vertex labels, and corresponding indexes in the tensor network:
In[6]:=
list=Developer`FromPackedArray@VertexList[net];TableForm[Transpose[Prepend[AnnotationValue[{net,list},#]&/@{"Tensor","Index"},Sort@AnnotationValue[net,VertexLabels]]],TableDepth2,TableHeadings{None,(Style[#1,Bold]&)/@{"Vertex index -> Vertex label","Tensor","Leg indices"}}]
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[7]:=
VertexList[net]
In[8]:=
Length@circuit["Flatten"]["Operators"]
In[9]:=
%Max[%%]
Note that each edge represents (i.e., is tagged by) a contraction:
In[10]:=
EdgeList[net]
Corresponding contraction:
In[11]:=
EdgeTags[net]
Perform the contraction:
In[12]:=
finalTensor=ContractTensorNetwork[net]
Confirm that the result is the same as default circuit application:
In[13]:=
circuit[]["Tensor"]finalTensor
Another tensor network representation uses indices as graph vertices with tensors as cliques:
In[14]:=
indexNet=TensorNetworkIndexGraph[net,GraphLayout{"LayeredDigraphEmbedding","Orientation"Left}]
In the above graph, the directed edges imply tensor contraction; also tensors are cliques in above graph:
In[15]:=
HighlightGraph[indexNet,Subgraph[indexNet,#]&/@FindClique[indexNet,Infinity,All]]
Free indices are the ones that left after contraction:
In[16]:=
TensorNetworkFreeIndices[net]
Highlight free indices:
In[17]:=
HighlightGraph[indexNet,TensorNetworkFreeIndices[net]]
Free indices can be extracted as vertices with zero in- and out- degree:
In[18]:=
Pick[VertexList[indexNet],VertexInDegree[indexNet]+VertexOutDegree[indexNet],0]
In[19]:=
ContainsExactly[TensorNetworkFreeIndices[net],TensorNetworkFreeIndices[net]]
Contraction and Einstein Summation
Contraction and Einstein Summation
Many useful information of a tensor network can be extracted using TensorNetworkData:
In[20]:=
TensorNetworkData[net]//Dataset
Also, one can get the useful data using graph functionalities, too:
In[21]:=
indices=AnnotationValue[{net,list},"Index"]/.Rule@@@EdgeTags[net]out=TensorNetworkFreeIndices[net]tensors=AnnotationValue[{net,list},"Tensor"];
Show ContractTensorNetwork is the same as EinsteinSummation:
In[22]:=
ContractTensorNetwork[net]===ActivateTensor@EinsteinSummation[indicesout,tensors]
One can compare the performance, on how the relevant computation is done
Perform the contraction in the order of network’s EdgeList:
In[23]:=
ContractTensorNetwork[net,Method"Naive"];//RepeatedTiming
Optimize the order for contraction, using EinsteinSummation and symbolic tensors package:
In[24]:=
ContractTensorNetwork[net];//RepeatedTiming
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:
In[25]:=
ψ0=
[{"RandomPure",3}];
| QuantumState |
Initialize the tensor network from above state:
In[26]:=
net2=
[{ψ0,circuit}]["TensorNetwork"];
 | QuantumCircuitOperator |
See supplement info, for package-scoped symbols.
Show the tensor contraction is the same as transformation of state by the circuit:
In[27]:=
ContractTensorNetwork[net2]circuit[ψ0]["Tensor"]
Supplement info
Supplement info
FromTensorNetwork
FromTensorNetwork
Any directed graph can be turned into a tensor network, even if the graph is not annotated.
In[28]:=
SeedRandom[1];graph=DirectedGraph[RandomGraph[{10,10}],"Acyclic",VertexLabelsv_HoldForm[v]]
Convert the graph into a tensor network such that tensors for each vertex is randomly generated in the dimension where the overall length of dimensions is determined by the tensor rank (overall legs/edges):
{2,2,…}
In[29]:=
qcTN=GraphTensorNetwork@graph
Get the tensor network data:
In[30]:=
TensorNetworkData@qcTN
One can assign symbolic tensor into vertices too:
In[31]:=
qcTNSymbolic=GraphTensorNetwork[graph,Method"Symbolic"];TensorNetworkData@qcTNSymbolic
Package-scoped symbols
Package-scoped symbols
In the Wolfram quantum framework, there are some package-scoped symbols that live in the context.
Wolfram`QuantumFramework`PackageScope`
In[32]:=
?Wolfram`QuantumFramework`PackageScope`$*
 |
|
""