This commit is contained in:
Giovanni Canali
2025-01-20 14:06:13 +01:00
committed by Nicola Demo
parent 81830ecc99
commit 08eaf56be1

View File

@@ -217,13 +217,13 @@ def laplacian(output_, input_, components=None, d=None, method="std"):
input_.shape[0], len(components), device=output_.device
)
labels = [None] * len(components)
for idx, ci in enumerate(components):
result[:, idx] = scalar_laplace(output_, input_, ci, d).flatten()
labels[idx] = f"dd{ci}"
for idx, c in enumerate(components):
result[:, idx] = scalar_laplace(output_, input_, c, d).flatten()
labels[idx] = f"dd{c}"
result = result.as_subclass(LabelTensor)
result.labels = labels
return result