Fast tensor rotation with NumPy Fast tensor rotation with NumPy numpy numpy