From 05e8d6765881b15059def90747e8f29b0e9de0bb Mon Sep 17 00:00:00 2001 From: Bi Tiekai Date: Fri, 24 May 2024 17:19:50 +0800 Subject: [PATCH] Update ndarray.py, fix the tile function here. --- python/needle/backend_ndarray/ndarray.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/needle/backend_ndarray/ndarray.py b/python/needle/backend_ndarray/ndarray.py index 48eb7e1..f66c164 100755 --- a/python/needle/backend_ndarray/ndarray.py +++ b/python/needle/backend_ndarray/ndarray.py @@ -508,7 +508,7 @@ def __matmul__(self, other): def tile(a, tile): return a.as_strided( (a.shape[0] // tile, a.shape[1] // tile, tile, tile), - (a.shape[1] * tile, tile, self.shape[1], 1), + (a.shape[1] * tile, tile, a.shape[1], 1), ) t = self.device.__tile_size__