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 input_.shape[0], len(components), device=output_.device
) )
labels = [None] * len(components) labels = [None] * len(components)
for idx, ci in enumerate(components): for idx, c in enumerate(components):
result[:, idx] = scalar_laplace(output_, input_, ci, d).flatten() result[:, idx] = scalar_laplace(output_, input_, c, d).flatten()
labels[idx] = f"dd{ci}" labels[idx] = f"dd{c}"
result = result.as_subclass(LabelTensor) result = result.as_subclass(LabelTensor)
result.labels = labels result.labels = labels
return result return result