folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🎨 Format Python code with psf/black (#333)

Browse Source 
Co-authored-by: ndem0 <ndem0@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2024-09-03 16:53:39 +02:00
committed by GitHub
parent eea0cc0833
commit 1aca017e1d
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pina/model/layers/orthogonal.py
View File

@@ -57,14 +57,12 @@ class OrthogonalBlock(torch.nn.Module):
v = torch.select(X, self.dim, i).clone()
for j in range(i):
vj = torch.select(result, self.dim, j).clone()
v = v - torch.sum(v * vj,
dim=self.dim, keepdim=True) * vj
v = v - torch.sum(v * vj, dim=self.dim, keepdim=True) * vj
# result_i = torch.select(result, self.dim, i)
result_i = v / torch.linalg.norm(v)
result = self._differentiable_copy(result, i, result_i)
return result
def _differentiable_copy(self, result, idx, value):
"""
Perform a differentiable copy operation on a tensor.
@@ -104,8 +102,10 @@ class OrthogonalBlock(torch.nn.Module):
# check consistency
check_consistency(value, int)
if value not in [0, 1, -1]:
raise IndexError('Dimension out of range (expected to be in '
f'range of [-1, 1], but got {value})')
raise IndexError(
"Dimension out of range (expected to be in "
f"range of [-1, 1], but got {value})"
)
# assign value
self._dim = value