Closed tiendung closed 2 years ago
https://github.com/dlsyscourse/hw3/blob/2403e58f0994852745b9456dc8c451068a35c8a9/python/needle/backend_ndarray/ndarray.py#L494
This should be (a.shape[1] * tile, tile, a.shape[1], 1),)
(a.shape[1] * tile, tile, a.shape[1], 1),)
https://github.com/dlsyscourse/hw3/blob/2403e58f0994852745b9456dc8c451068a35c8a9/python/needle/backend_ndarray/ndarray.py#L494
This should be
(a.shape[1] * tile, tile, a.shape[1], 1),)