pytorch

1
import torch
2
from typing import TypeVar
3

4
T = TypeVar('T')
5

6
# returns if all are the same mode
7
def all_same_mode(modes):
8
    return all(tuple(mode == modes[0] for mode in modes))
9

10
no_dispatch = torch._C._DisableTorchDispatch
11