torch.func.rearrange ==================== .. currentmodule:: torch.func .. autofunction:: rearrange