def_make_rand_tensor(shape:Shape,low:float=0.,high:float=1.,
- device:torch.device=torch.device('cpu'))->Tensor:
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->Tensor:"""Returns tensor whose entries are drawn from the uniform distribution."""ifnotisinstance(low,float):raiseTypeError('`low` should be float type')
@@ -1520,13 +1556,14 @@
Source code for tensorkrowch.components
raiseTypeError('`high` should be float type')iflow>=high:raiseValueError('`low` should be strictly smaller than `high`')
- returntorch.rand(shape,device=device)*(high-low)+low
+ returntorch.rand(shape,device=device,dtype=dtype)*(high-low)+low@staticmethoddef_make_randn_tensor(shape:Shape,mean:float=0.,std:float=1.,
- device:torch.device=torch.device('cpu'))->Tensor:
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->Tensor:"""Returns tensor whose entries are drawn from the normal distribution."""ifnotisinstance(mean,float):raiseTypeError('`mean` should be float type')
@@ -1534,12 +1571,13 @@
Source code for tensorkrowch.components
raiseTypeError('`std` should be float type')ifstd<=0:raiseValueError('`std` should be positive')
- returntorch.randn(shape,device=device)*std+mean
+ returntorch.randn(shape,device=device,dtype=dtype)*std+mean
[docs]defmake_tensor(self,shape:Optional[Shape]=None,init_method:Text='zeros',
- device:torch.device=torch.device('cpu'),
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->Tensor:""" Returns a tensor that can be put in the node, and is initialized according
@@ -1553,6 +1591,8 @@
Source code for tensorkrowch.components
Initialization method. device : torch.device, optional Device where to initialize the tensor.
+ dtype : torch.dtype, optional
+ Dtype of the tensor. kwargs : float Keyword arguments for the different initialization methods:
@@ -1574,15 +1614,17 @@
tensor:Optional[Tensor]=None,init_method:Optional[Text]='zeros',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Sets a new node's tensor or creates one with :meth:`make_tensor` and sets
@@ -1673,6 +1716,8 @@
Source code for tensorkrowch.components
Initialization method. device : torch.device, optional Device where to initialize the tensor.
+ dtype : torch.dtype, optional
+ Dtype of the tensor. kwargs : float Keyword arguments for the different initialization methods. See :meth:`make_tensor`.
@@ -1680,10 +1725,14 @@
Source code for tensorkrowch.components
iftensorisnotNone:ifnotisinstance(tensor,Tensor):raiseTypeError('`tensor` should be torch.Tensor type')
- elifdeviceisnotNone:
+ ifdeviceisnotNone:warnings.warn('`device` was specified but is being ignored. ''Provide a tensor that is already in the required'' device')
+ ifdtypeisnotNone:
+ warnings.warn('`dtype` was specified but is being ignored. '
+ 'Provide a tensor that already has the required'
+ ' dtype')ifnotself._compatible_shape(tensor):tensor=self._crop_tensor(tensor)
@@ -1691,10 +1740,13 @@
tensor:Optional[Tensor]=None,init_method:Optional[Text]='zeros',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Sets new node's tensor or creates one with :meth:`make_tensor` and sets
@@ -1740,6 +1793,8 @@
Source code for tensorkrowch.components
Initialization method. device : torch.device, optional Device where to initialize the tensor.
+ dtype : torch.dtype, optional
+ Dtype of the tensor. kwargs : float Keyword arguments for the different initialization methods. See :meth:`make_tensor`.
@@ -1780,6 +1835,7 @@
Source code for tensorkrowch.components
self._unrestricted_set_tensor(tensor=tensor,init_method=init_method,device=device,
+ dtype=dtype,**kwargs)else:raiseValueError('Node\'s tensor can only be changed if it is not'
@@ -2332,6 +2388,8 @@
Source code for tensorkrowch.components
Initialization method. device : torch.device, optional Device where to initialize the tensor if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`AbstractNode.make_tensor`.
@@ -2656,6 +2714,8 @@
Source code for tensorkrowch.components
Initialization method. device : torch.device, optional Device where to initialize the tensor if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`AbstractNode.make_tensor`.
@@ -2737,6 +2797,7 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -448,6 +450,7 @@
tensors:Optional[Sequence[torch.Tensor]]=None,init_method:Optional[Text]='randn',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Initializes all the nodes of the :class:`MPO`. It can be called when
@@ -735,7 +740,7 @@
Source code for tensorkrowch.models.mpo
* ``{"zeros", "ones", "copy", "rand", "randn"}``: Each node is initialized calling :meth:`~tensorkrowch.AbstractNode.set_tensor` with
- the given method, ``device`` and ``kwargs``.
+ the given method, ``device``, ``dtype`` and ``kwargs``. Parameters ----------
@@ -749,6 +754,8 @@
Source code for tensorkrowch.models.mpo
Initialization method. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -763,6 +770,8 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -1380,6 +1401,7 @@
tensors:Optional[Sequence[torch.Tensor]]=None,init_method:Optional[Text]='randn',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Initializes the common tensor of the :class:`UMPO`. It can be called
@@ -1458,7 +1482,7 @@
Source code for tensorkrowch.models.mpo
* ``{"zeros", "ones", "copy", "rand", "randn"}``: The tensor is initialized calling :meth:`~tensorkrowch.AbstractNode.set_tensor` with
- the given method, ``device`` and ``kwargs``.
+ the given method, ``device``, ``dtype`` and ``kwargs``. Parameters ----------
@@ -1470,6 +1494,8 @@
Source code for tensorkrowch.models.mpo
Initialization method. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -1480,6 +1506,7 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -497,6 +499,7 @@
ifi==self._n_features-1:self._mats_env[-1]['right']^self._right_node['left']
- def_make_canonical(self,device:Optional[torch.device]=None)->List[torch.Tensor]:
+ def_make_canonical(self,
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->List[torch.Tensor]:""" Creates random unitaries to initialize the MPS in canonical form with orthogonality center at the rightmost node. Unitaries in nodes are
@@ -934,22 +940,22 @@
Source code for tensorkrowch.models.mps
phys_dim=node_shape[1]size=max(aux_shape[0],aux_shape[1])
- tensor=random_unitary(size,device=device)
+ tensor=random_unitary(size,device=device,dtype=dtype)tensor=tensor[:min(aux_shape[0],size),:min(aux_shape[1],size)]tensor=tensor.reshape(*node_shape)ifi==(self._n_features-1):
- ifself._boundary=='obc':
- tensor=tensor.t()/tensor.norm()*sqrt(phys_dim)
- else:
- tensor=tensor/tensor.norm()*sqrt(phys_dim)
- else:
- tensor=tensor*sqrt(phys_dim)
+ if(self._boundary=='obc')and(i==0):
+ tensor=tensor[:,0]
+ tensor=tensor/tensor.norm()
+ tensor=tensor*sqrt(phys_dim)tensors.append(tensor)returntensors
- def_make_unitaries(self,device:Optional[torch.device]=None)->List[torch.Tensor]:
+ def_make_unitaries(self,
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->List[torch.Tensor]:""" Creates random unitaries to initialize the MPS nodes as stacks of unitaries.
@@ -973,7 +979,7 @@
[docs]definitialize(self,tensors:Optional[Sequence[torch.Tensor]]=None,init_method:Optional[Text]='randn',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Initializes all the nodes of the :class:`MPS`. It can be called when
@@ -1003,7 +1008,7 @@
Source code for tensorkrowch.models.mps
* ``{"zeros", "ones", "copy", "rand", "randn"}``: Each node is initialized calling :meth:`~tensorkrowch.AbstractNode.set_tensor` with
- the given method, ``device`` and ``kwargs``.
+ the given method, ``device``, ``dtype`` and ``kwargs``. * ``"randn_eye"``: Nodes are initialized as in this `paper <https://arxiv.org/abs/1605.03795>`_, adding identities at the
@@ -1033,14 +1038,16 @@
Source code for tensorkrowch.models.mps
Initialization method. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`. """ifinit_method=='unit':
- tensors=self._make_unitaries(device=device)
+ tensors=self._make_unitaries(device=device,dtype=dtype)elifinit_method=='canonical':
- tensors=self._make_canonical(device=device)
+ tensors=self._make_canonical(device=device,dtype=dtype)iftensorsisnotNone:iflen(tensors)!=self._n_features:
@@ -1052,19 +1059,23 @@
side:Text='right'):"""Projects all nodes into a space of dimension ``bond_dim``."""device=nodes[0].tensor.device
+ dtype=nodes[0].tensor.dtypeifside=='left':nodes.reverse()
@@ -2219,7 +2256,7 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -2452,6 +2492,7 @@
fornodeinself._mats_env:node.set_tensor_from(uniform_memory)
- def_make_canonical(self,device:Optional[torch.device]=None)->List[torch.Tensor]:
+ def_make_canonical(self,
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->List[torch.Tensor]:""" Creates random unitaries to initialize the MPS in canonical form with orthogonality center at the rightmost node. Unitaries in nodes are
@@ -2527,13 +2571,15 @@
Source code for tensorkrowch.models.mps
size=max(aux_shape[0],aux_shape[1])phys_dim=node_shape[1]
- tensor=random_unitary(size,device=device)
+ tensor=random_unitary(size,device=device,dtype=dtype)tensor=tensor[:min(aux_shape[0],size),:min(aux_shape[1],size)]tensor=tensor.reshape(*node_shape)tensor=tensor*sqrt(phys_dim)returntensor
- def_make_unitaries(self,device:Optional[torch.device]=None)->List[torch.Tensor]:
+ def_make_unitaries(self,
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->List[torch.Tensor]:""" Creates random unitaries to initialize the MPS nodes as stacks of unitaries.
@@ -2543,7 +2589,7 @@
tensors:Optional[Sequence[torch.Tensor]]=None,init_method:Optional[Text]='randn',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Initializes the common tensor of the :class:`UMPS`. It can be called
@@ -2561,7 +2608,7 @@
Source code for tensorkrowch.models.mps
* ``{"zeros", "ones", "copy", "rand", "randn"}``: The tensor is initialized calling :meth:`~tensorkrowch.AbstractNode.set_tensor` with
- the given method, ``device`` and ``kwargs``.
+ the given method, ``device``, ``dtype`` and ``kwargs``. * ``"randn_eye"``: Tensor is initialized as in this `paper <https://arxiv.org/abs/1605.03795>`_, adding identities at the
@@ -2588,6 +2635,8 @@
Source code for tensorkrowch.models.mps
Initialization method. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -2595,9 +2644,9 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -2829,6 +2883,7 @@
"""Returns the output node."""returnself._mats_env[self._out_position]
- def_make_canonical(self,device:Optional[torch.device]=None)->List[torch.Tensor]:
+ def_make_canonical(self,
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->List[torch.Tensor]:""" Creates random unitaries to initialize the MPS in canonical form with orthogonality center at the rightmost node. Unitaries in nodes are
@@ -2944,14 +3002,16 @@
tensors=left_tensors+[out_tensor]+right_tensorsreturntensors
- def_make_unitaries(self,device:Optional[torch.device]=None)->List[torch.Tensor]:
+ def_make_unitaries(self,
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->List[torch.Tensor]:""" Creates random unitaries to initialize the MPS nodes as stacks of unitaries.
@@ -3011,7 +3073,7 @@
# All tensorstensors=left_tensors+[out_tensor]+right_tensorsreturntensors
-
- definitialize(self,
- tensors:Optional[Sequence[torch.Tensor]]=None,
- init_method:Optional[Text]='randn',
- device:Optional[torch.device]=None,
- **kwargs:float)->None:
-"""
- Initializes all the nodes of the :class:`MPS`. It can be called when
- instantiating the model, or to override the existing nodes' tensors.
-
- There are different methods to initialize the nodes:
-
- * ``{"zeros", "ones", "copy", "rand", "randn"}``: Each node is
- initialized calling :meth:`~tensorkrowch.AbstractNode.set_tensor` with
- the given method, ``device`` and ``kwargs``.
-
- * ``"randn_eye"``: Nodes are initialized as in this
- `paper <https://arxiv.org/abs/1605.03795>`_, adding identities at the
- top of random gaussian tensors. In this case, ``std`` should be
- specified with a low value, e.g., ``std = 1e-9``.
-
- * ``"unit"``: Nodes are initialized as stacks of random unitaries. This,
- combined (at least) with an embedding of the inputs as elements of
- the computational basis (:func:`~tensorkrowch.embeddings.discretize`
- combined with :func:`~tensorkrowch.embeddings.basis`)
-
- * ``"canonical"```: MPS is initialized in canonical form with a squared
- norm `close` to the product of all the physical dimensions (if bond
- dimensions are bigger than the powers of the physical dimensions,
- the norm could vary). Th orthogonality center is at the rightmost
- node.
-
- Parameters
- ----------
- tensors : list[torch.Tensor] or tuple[torch.Tensor], optional
- Sequence of tensors to set in each of the MPS nodes. If ``boundary``
- is ``"obc"``, all tensors should be rank-3, except the first and
- last ones, which can be rank-2, or rank-1 (if the first and last are
- the same). If ``boundary`` is ``"pbc"``, all tensors should be
- rank-3.
- init_method : {"zeros", "ones", "copy", "rand", "randn", "randn_eye", "unit", "canonical"}, optional
- Initialization method.
- device : torch.device, optional
- Device where to initialize the tensors if ``init_method`` is provided.
- kwargs : float
- Keyword arguments for the different initialization methods. See
- :meth:`~tensorkrowch.AbstractNode.make_tensor`.
- """
- ifinit_method=='unit':
- tensors=self._make_unitaries(device=device)
- elifinit_method=='canonical':
- tensors=self._make_canonical(device=device)
[docs]definitialize(self,tensors:Optional[Sequence[torch.Tensor]]=None,init_method:Optional[Text]='randn',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Initializes all the nodes of the :class:`MPSLayer`. It can be called when
@@ -3134,7 +3147,7 @@
Source code for tensorkrowch.models.mps
* ``{"zeros", "ones", "copy", "rand", "randn"}``: Each node is initialized calling :meth:`~tensorkrowch.AbstractNode.set_tensor` with
- the given method, ``device`` and ``kwargs``.
+ the given method, ``device``, ``dtype`` and ``kwargs``. * ``"randn_eye"``: Nodes are initialized as in this `paper <https://arxiv.org/abs/1605.03795>`_, adding identities at the
@@ -3163,14 +3176,16 @@
Source code for tensorkrowch.models.mps
Initialization method. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`. """ifinit_method=='unit':
- tensors=self._make_unitaries(device=device)
+ tensors=self._make_unitaries(device=device,dtype=dtype)elifinit_method=='canonical':
- tensors=self._make_canonical(device=device)
+ tensors=self._make_canonical(device=device,dtype=dtype)iftensorsisnotNone:iflen(tensors)!=self._n_features:
@@ -3182,19 +3197,23 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -3358,6 +3388,7 @@
fornodeinin_nodes:node.set_tensor_from(uniform_memory)
- def_make_canonical(self,device:Optional[torch.device]=None)->List[torch.Tensor]:
+ def_make_canonical(self,
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->List[torch.Tensor]:""" Creates random unitaries to initialize the MPS in canonical form with orthogonality center at the rightmost node. Unitaries in nodes are
@@ -3496,18 +3530,22 @@
Source code for tensorkrowch.models.mps
size=max(aux_shape[0],aux_shape[1])phys_dim=node_shape[1]
- uni_tensor=random_unitary(size,device=device)
+ uni_tensor=random_unitary(size,device=device,dtype=dtype)uni_tensor=uni_tensor[:min(aux_shape[0],size),:min(aux_shape[1],size)]uni_tensor=uni_tensor.reshape(*node_shape)uni_tensor=uni_tensor*sqrt(phys_dim)# Output node
- out_tensor=torch.randn(self.out_node.shape,device=device)
+ out_tensor=torch.randn(self.out_node.shape,
+ device=device,
+ dtype=dtype)out_tensor=out_tensor/out_tensor.norm()*sqrt(out_tensor.shape[1])return[uni_tensor,out_tensor]
- def_make_unitaries(self,device:Optional[torch.device]=None)->List[torch.Tensor]:
+ def_make_unitaries(self,
+ device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None)->List[torch.Tensor]:""" Creates random unitaries to initialize the MPS nodes as stacks of unitaries.
@@ -3518,7 +3556,9 @@
tensors:Optional[Sequence[torch.Tensor]]=None,init_method:Optional[Text]='randn',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Initializes the common tensor of the :class:`UMPSLayer`. It can be called
@@ -3538,7 +3579,7 @@
Source code for tensorkrowch.models.mps
* ``{"zeros", "ones", "copy", "rand", "randn"}``: The tensor is initialized calling :meth:`~tensorkrowch.AbstractNode.set_tensor` with
- the given method, ``device`` and ``kwargs``.
+ the given method, ``device``, ``dtype`` and ``kwargs``. * ``"randn_eye"``: Tensor is initialized as in this `paper <https://arxiv.org/abs/1605.03795>`_, adding identities at the
@@ -3566,14 +3607,16 @@
Source code for tensorkrowch.models.mps
Initialization method. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`. """ifinit_method=='unit':
- tensors=self._make_unitaries(device=device)
+ tensors=self._make_unitaries(device=device,dtype=dtype)elifinit_method=='canonical':
- tensors=self._make_canonical(device=device)
+ tensors=self._make_canonical(device=device,dtype=dtype)iftensorsisnotNone:self.uniform_memory.tensor=tensors[0]
@@ -3588,12 +3631,14 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -3895,6 +3943,7 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -4061,6 +4114,7 @@
explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -4241,6 +4299,7 @@
detailed explanation of the different initialization methods. device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -4429,6 +4492,7 @@
:meth:`add_data` to see how to initialize MPS nodes with data tensors.
device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`.
@@ -439,6 +441,7 @@
"""Returns the list of nodes in ``mats_env``."""returnself._mats_env
+ @property
+ deftensors(self)->List[torch.Tensor]:
+"""Returns the list of MPS tensors."""
+ mps_tensors=[node.tensorfornodeinself._mats_env]
+ ifself._boundary=='obc':
+ mps_tensors[0]=mps_tensors[0][0,:,:]
+ mps_tensors[-1]=mps_tensors[-1][:,:,0]
+ returnmps_tensors
+
# -------# Methods# -------
@@ -673,6 +686,7 @@
Source code for tensorkrowch.models.mps_data
[docs]definitialize(self,init_method:Optional[Text]='randn',device:Optional[torch.device]=None,
+ dtype:Optional[torch.dtype]=None,**kwargs:float)->None:""" Initializes all the nodes of the :class:`MPSData`. It can be called when
@@ -682,7 +696,7 @@
Source code for tensorkrowch.models.mps_data
* ``{"zeros", "ones", "copy", "rand", "randn"}``: Each node is initialized calling :meth:`~tensorkrowch.AbstractNode.set_tensor` with
- the given method, ``device`` and ``kwargs``.
+ the given method, ``device``, ``dtype`` and ``kwargs``. Parameters ----------
@@ -691,22 +705,31 @@
Source code for tensorkrowch.models.mps_data
initialize MPS nodes with data tensors.
device : torch.device, optional Device where to initialize the tensors if ``init_method`` is provided.
+ dtype : torch.dtype, optional
+ Dtype of the tensor if ``init_method`` is provided. kwargs : float Keyword arguments for the different initialization methods. See :meth:`~tensorkrowch.AbstractNode.make_tensor`. """ifself._boundary=='obc':
- self._left_node.set_tensor(init_method='copy',device=device)
- self._right_node.set_tensor(init_method='copy',device=device)
+ self._left_node.set_tensor(init_method='copy',
+ device=device,
+ dtype=dtype)
+ self._right_node.set_tensor(init_method='copy',
+ device=device,
+ dtype=dtype)ifinit_methodisnotNone:fori,nodeinenumerate(self._mats_env):node.set_tensor(init_method=init_method,device=device,
+ dtype=dtype,**kwargs)ifself._boundary=='obc':
- aux_tensor=torch.zeros(*node.shape,device=device)
+ aux_tensor=torch.zeros(*node.shape,
+ device=device,
+ dtype=dtype)ifi==0:# Left nodeaux_tensor[...,0,:,:]=node.tensor[...,0,:,:]
@@ -762,37 +785,45 @@