We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0af4221 commit 401ea8eCopy full SHA for 401ea8e
1 file changed
backends/arm/process_node.py
@@ -42,6 +42,8 @@ def _tensor_to_numpy_with_dim_order(
42
np_tensor = tensor.view(torch.uint16).numpy().view(ml_dtypes.bfloat16)
43
else:
44
np_tensor = tensor.numpy()
45
+ if dim_order == tuple(range(len(dim_order))):
46
+ return np_tensor
47
return np.transpose(np_tensor, dim_order)
48
49
0 commit comments