S{N} can be a number.

2026-02-14 11:42:26 +01:00
parent 0817e50109
commit 5dabd12c9a
1 changed files with 1 additions and 1 deletions
@@ -65,7 +65,7 @@ def generate_dim_variables(tensor: torch.Tensor):
    variables = {}
    for dim, size in enumerate(tensor.shape):
        variables[f"D{dim}"] = getIndexTensorAlongDim(tensor, dim)
-        variables[f"S{dim}"] = torch.full(tensor.shape, fill_value=size, dtype=torch.float32, device=tensor.device)
+        variables[f"S{dim}"] = float(size)  # Scalar size, not tensor
    return variables