pytorch/torch/_C/_VariableFunctions.pyi.in
Chung-chieh Shan 2c588b3ad5 Allow new_full's fill_value argument type to be complex (#91345)
It seems that this code should type-check but doesn't:
```python
torch.zeros((2,3),dtype=torch.cdouble).new_full((4,5),complex(6,7))
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/91345
Approved by: https://github.com/zou3519, https://github.com/ezyang
2023-03-21 12:34:00 +00:00

40 lines
545 B
Python

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