1 # ${generated_comment} 2 # mypy: disable-error-code="type-arg" 3 # mypy: allow-untyped-defs 4 5 import builtins 6 from typing import ( 7 Any, 8 Callable, 9 ContextManager, 10 Iterator, 11 List, 12 Literal, 13 NamedTuple, 14 Optional, 15 overload, 16 Sequence, 17 Tuple, 18 TypeVar, 19 Union, 20 ) 21 22 import torch 23 from torch import contiguous_format, Generator, inf, memory_format, strided, SymInt, Tensor 24 from torch.types import ( 25 _bool, 26 _complex, 27 _device, 28 _dtype, 29 _float, 30 _int, 31 _layout, 32 _qscheme, 33 _size, 34 Device, 35 Number, 36 ) 37 38 from torch._prims_common import DeviceLikeType 39 40 ${function_hints} 41 42 ${all_directive} 43