mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
71003c74f8
Currently, `NamedTuple` return types are created in `torch/_VF.pyi` instead of typing being added for the symbols in in `torch/return_types.py`. This also fixes the type names to match the actual names in the python code. Pull Request resolved: https://github.com/pytorch/pytorch/pull/74199 Approved by: https://github.com/ezyang
16 lines
470 B
Python
16 lines
470 B
Python
# ${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
|
|
import torch
|
|
|
|
import builtins
|
|
|
|
${function_hints}
|
|
|
|
${all_directive}
|