diff --git a/torchvision/datasets/mnist.py b/torchvision/datasets/mnist.py index edd2185c984..0467854e9b3 100644 --- a/torchvision/datasets/mnist.py +++ b/torchvision/datasets/mnist.py @@ -496,7 +496,7 @@ def read_sn3_pascalvincent_tensor(path: str, strict: bool = True) -> torch.Tenso s = [get_int(data[4 * (i + 1): 4 * (i + 2)]) for i in range(nd)] parsed = np.frombuffer(data, dtype=m[1], offset=(4 * (nd + 1))) assert parsed.shape[0] == np.prod(s) or not strict - return torch.from_numpy(parsed.astype(m[2], copy=False)).view(*s) + return torch.from_numpy(parsed.astype(m[2])).view(*s) def read_label_file(path: str) -> torch.Tensor: