remove debug prints

This commit is contained in:
mcDandy
2025-12-30 15:32:52 +01:00
parent 47f4d1309d
commit a9322ab055
-5
View File
@@ -91,11 +91,6 @@ def generate_dim_variables(tensor):
variables = {}
for dim, size in enumerate(tensor.shape):
variables[f'D{dim}'] = getIndexTensorAlongDim(tensor, dim)
print(tensor)
print(tensor.shape)
print(variables[f'D{dim}'])
print(dim)
print(size)
variables[f'S{dim}'] = torch.full(tensor.shape, fill_value=size, dtype=torch.float32, device=tensor.device)
return variables