1# ${generated_comment} 2# mypy: disable-error-code="type-arg" 3# mypy: allow-untyped-defs 4 5import builtins 6from 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 22import torch 23from torch import contiguous_format, Generator, inf, memory_format, strided, SymInt, Tensor 24from 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 38from torch._prims_common import DeviceLikeType 39 40${function_hints} 41 42${all_directive} 43