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

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

Browse Source 
Co-authored-by: dario-coscia <dario-coscia@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2024-11-18 17:35:15 +01:00
committed by GitHub
parent a78f44ecef
commit b896a2a074
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pina/operators.py
View File

@@ -184,7 +184,7 @@ def laplacian(output_, input_, components=None, d=None, method="std"):
Laplacian. It has to be a column tensor. Laplacian. It has to be a column tensor.
:param LabelTensor input_: the input tensor with respect to which :param LabelTensor input_: the input tensor with respect to which
computing the Laplacian. computing the Laplacian.
:param list(str) components: the name of the output variables to :param list(str) components: the name of the output variables to
calculate the Laplacian for. It should be a subset of the output calculate the Laplacian for. It should be a subset of the output
labels. If None, all the output variables are considered. labels. If None, all the output variables are considered.
:param list(str) d: the name of the input variables on which the :param list(str) d: the name of the input variables on which the
@@ -224,7 +224,8 @@ def laplacian(output_, input_, components=None, d=None, method="std"):
else: else:
result = torch.empty( result = torch.empty(
size=(input_.shape[0], len(components)), size=(input_.shape[0], len(components)),
dtype=output_.dtype, device=output_.device dtype=output_.dtype,
device=output_.device,
) )
labels = [None] * len(components) labels = [None] * len(components)
for idx, c in enumerate(components): for idx, c in enumerate(components):