dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed Bug with Givens Rotation (Eigen)

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-10-09 11:24:06 +02:00
parent f9a08add40
commit e31c25135e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
metastable_baselines/distributions/distributions.py
View File

@ -594,7 +594,7 @@ class CholNet(nn.Module):
eigenv = th.cumsum(eigenv, -1) eigenv = th.cumsum(eigenv, -1)
# reverse order, oh well... # reverse order, oh well...
self._givens_rot.theta = theta self._givens_rotator.theta = theta
Q = self._givens_rotator(self._givens_ident) Q = self._givens_rotator(self._givens_ident)
Qinv = Q.transpose(dim0=-2, dim1=-1) Qinv = Q.transpose(dim0=-2, dim1=-1)