Tensor

[1]:

from lambeq import BobcatParser

parser = BobcatParser()
diagram = parser.sentence2diagram('Alice gives Claire the flowers')
diagram.draw()
../_images/examples_tensor_1_0.png 
[2]:

from lambeq import AtomicType, MPSAnsatz, SpiderAnsatz, TensorAnsatz
from discopy import Dim

N = AtomicType.NOUN
S = AtomicType.SENTENCE

[3]:

TensorAnsatz({N: Dim(4), S: Dim(3)})(diagram).draw()
../_images/examples_tensor_3_0.png 
[4]:

MPSAnsatz({N: Dim(4), S: Dim(3)}, bond_dim=5)(diagram).draw(figsize=(8, 4))
../_images/examples_tensor_4_0.png 
[5]:

SpiderAnsatz({N: Dim(4), S: Dim(3)})(diagram).draw()
../_images/examples_tensor_5_0.png 
[6]:

import torch as th
from sympy import default_sort_key

d = SpiderAnsatz({N: Dim(4), S: Dim(3)}, max_order=2)(diagram)

syms = sorted(d.free_symbols, key=default_sort_key)
sym_dict = {k: th.ones(k.size) for k in syms}
subbed_diagram = d.lambdify(*syms)(*sym_dict.values())

[7]:

import tensornetwork as tn
subbed_diagram.eval(contractor=tn.contractors.auto)

[7]:

Tensor(dom=Dim(1), cod=Dim(3), array=[256., 256., 256.])