Make grid float instead of int

jloveric · Jun 23, 2024 · 5722b5c · 5722b5c
1 parent b122a2f
commit 5722b5c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/block_mnist.py b/examples/block_mnist.py
@@ -33,10 +33,10 @@
 }
 
 grid_x, grid_y = torch.meshgrid(
-    (torch.arange(28) - 14) // 14, (torch.arange(28) - 14) // 14, indexing="ij"
+    (torch.arange(28) - 14) / 14, (torch.arange(28) - 14) / 14, indexing="ij"
 )
 grid = torch.stack([grid_x, grid_y])
-
+print('grid', grid)
 
 def collate_fn(batch):