| # ${generated_comment} | |
| from torch import Tensor, Generator, strided, memory_format, contiguous_format, strided | |
| from typing import List, Tuple, Optional, Union, Any, ContextManager, Callable, overload, Iterator, NamedTuple, Sequence, TypeVar | |
| from typing_extensions import Literal | |
| from torch._six import inf | |
| from torch.types import _int, _float, _bool, Number, _dtype, _device, _qscheme, _size, _layout, SymInt, Device | |
| import torch | |
| import builtins | |
| ${function_hints} | |
| ${all_directive} |