1. 找到torch安装目录

  • D:\ProgramData\Anaconda3\envs\study-ml\Lib\site-packages\torch
  • 我的路径在,conda的虚拟环境下的study-ml下
  • 找到这个路径:Lib\site-packages\torch
  • 版本大于1.4以后似乎就没有initpyi这个文件了
  • 下载init.pyi附件,放在上面的路径下就可以了

    2. init.pyi下载地址

    网盘地址:https://pan.baidu.com/s/1DUcJq-fj0VR3xUiYkXHBA?_at=1622265991308
    提取码:6pwe

原因是什么没有摸清,但是基本上可以确定是由于init.pyi缺失或者存在问题导致的。我的pytorch版本是0.4.1,但是在网上没有找到0.4.1的相关init.pyi的版本,所以我使用的是1.0.1的。

3. 解决方法:

进入到这个链接中,在pytorch文件夹下,找到一个版本的init.pyi放到你的虚拟环境所在目录下的torch中。如我的torch包所在位置为D:\Program Files (x86)\Anaconda\envs\pytorch\Lib\site-packages\torch,然后将init.pyi复制进来
到这就结束了吗?并没有,当你重启运行pycharm时,可以自动补全torch.sum、torch.mean等函数,但是却无法引用torch.nn等模块,此时只需要在init.pyi文件中加入下面两行即可。

  1. from torch import nn, cuda, ops, functional, optim, autograd, onnx, utils
  2. from torch import contrib, distributions, for_onnx, jit, multiprocessing

加入前:
pytorch在pycharm中没有代码自动提示 - 图1
加入后:
pytorch在pycharm中没有代码自动提示 - 图2
然后,再重启pycharm即可。
pytorch在pycharm中没有代码自动提示 - 图3
可以看到,代码自动补全功能已经恢复了。
pytorch在pycharm中没有代码自动提示 - 图4

版权声明:本文为qq_35531985原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qq_35531985/article/details/107980736

init.pyi文件可直接复制

  1. from typing import List, Tuple, Optional, Union, Any, ContextManager, Callable, overload
  2. import builtins
  3. import math
  4. import pickle
  5. from torch import nn, cuda, ops, functional, optim, autograd, onnx, utils
  6. from torch import contrib, distributions, for_onnx, jit, multiprocessing
  7. class dtype: ...
  8. _dtype = dtype
  9. class layout: ...
  10. strided : layout = ...
  11. class device:
  12. def __init__(self, device: Union[device, str, None]=None) -> None: ...
  13. class Generator: ...
  14. class Size(tuple): ...
  15. class Storage: ...
  16. class enable_grad():
  17. def __enter__(self) -> None: ...
  18. def __exit__(self, *args) -> None: ...
  19. def __call__(self, func : Callable) -> Callable: ...
  20. class no_grad():
  21. def __enter__(self) -> None: ...
  22. def __exit__(self, *args) -> None: ...
  23. def __call__(self, func : Callable) -> Callable: ...
  24. class set_grad_enabled():
  25. def __init__(self, mode: bool) -> None: ...
  26. def __enter__(self) -> None: ...
  27. def __exit__(self, *args) -> None: ...
  28. class Tensor:
  29. dtype: _dtype = ...
  30. shape: Size = ...
  31. requires_grad: bool = ...
  32. grad: Optional['Tensor'] = ...
  33. def __abs__(self) -> 'Tensor': ...
  34. def __add__(self, other: Any) -> 'Tensor': ...
  35. def __and__(self, other: Any) -> 'Tensor': ...
  36. def __array__(self, dtype=None): ...
  37. def __array_wrap__(self, array): ...
  38. def __bool__(self) -> bool: ...
  39. def __deepcopy__(self, memo): ...
  40. def __dir__(self): ...
  41. def __div__(self, other: Any) -> 'Tensor': ...
  42. def __eq__(self, other: Any) -> 'Tensor': ... # type: ignore
  43. def __float__(self) -> builtins.float: ...
  44. def __floordiv__(self, other): ...
  45. def __format__(self, format_spec): ...
  46. def __ge__(self, other: Any) -> 'Tensor': ... # type: ignore
  47. def __getitem__(self, indices: Union[None, builtins.int, slice, 'Tensor', List, Tuple]) -> 'Tensor': ...
  48. def __gt__(self, other: Any) -> 'Tensor': ... # type: ignore
  49. def __hash__(self): ...
  50. def __iadd__(self, other: Any) -> 'Tensor': ...
  51. def __iand__(self, other: Any) -> 'Tensor': ...
  52. def __idiv__(self, other: Any) -> 'Tensor': ...
  53. def __ilshift__(self, other: Any) -> 'Tensor': ...
  54. def __imul__(self, other: Any) -> 'Tensor': ...
  55. def __index__(self) -> builtins.int: ...
  56. def __int__(self) -> builtins.int: ...
  57. def __invert__(self) -> 'Tensor': ...
  58. def __ior__(self, other: Any) -> 'Tensor': ...
  59. def __ipow__(self, other): ...
  60. def __irshift__(self, other: Any) -> 'Tensor': ...
  61. def __isub__(self, other: Any) -> 'Tensor': ...
  62. def __iter__(self): ...
  63. def __itruediv__(self, other: Any) -> 'Tensor': ...
  64. def __ixor__(self, other: Any) -> 'Tensor': ...
  65. def __le__(self, other: Any) -> 'Tensor': ... # type: ignore
  66. def __len__(self): ...
  67. def __long__(self) -> builtins.int: ...
  68. def __lshift__(self, other: Any) -> 'Tensor': ...
  69. def __lt__(self, other: Any) -> 'Tensor': ... # type: ignore
  70. def __matmul__(self, other: Any) -> 'Tensor': ...
  71. def __mod__(self, other: Any) -> 'Tensor': ...
  72. def __mul__(self, other: Any) -> 'Tensor': ...
  73. def __ne__(self, other: Any) -> 'Tensor': ... # type: ignore
  74. def __neg__(self) -> 'Tensor': ...
  75. def __nonzero__(self) -> bool: ...
  76. def __or__(self, other: Any) -> 'Tensor': ...
  77. def __pow__(self, other: Any) -> 'Tensor': ...
  78. def __radd__(self, other: Any) -> 'Tensor': ...
  79. def __rdiv__(self, other): ...
  80. def __reduce_ex__(self, proto): ...
  81. def __repr__(self): ...
  82. def __reversed__(self): ...
  83. def __rfloordiv__(self, other): ...
  84. def __rmul__(self, other: Any) -> 'Tensor': ...
  85. def __rpow__(self, other): ...
  86. def __rshift__(self, other: Any) -> 'Tensor': ...
  87. def __rsub__(self, other): ...
  88. def __rtruediv__(self, other): ...
  89. def __setitem__(self, indices: Union[None, builtins.int, slice, 'Tensor', List, Tuple], val: Union['Tensor', builtins.float, builtins.int]) -> None: ...
  90. def __setstate__(self, state): ...
  91. def __sub__(self, other: Any) -> 'Tensor': ...
  92. def __truediv__(self, other: Any) -> 'Tensor': ...
  93. def __xor__(self, other: Any) -> 'Tensor': ...
  94. def abs(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  95. def abs_(self) -> 'Tensor': ...
  96. def acos(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  97. def acos_(self) -> 'Tensor': ...
  98. @overload
  99. def add(self, other: 'Tensor', *, alpha: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  100. @overload
  101. def add(self, other: Union[builtins.float, builtins.int], alpha: Union[builtins.float, builtins.int]=1, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  102. @overload
  103. def add_(self, other: 'Tensor', *, alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  104. @overload
  105. def add_(self, other: Union[builtins.float, builtins.int], alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  106. def addbmm(self, batch1: 'Tensor', batch2: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  107. def addbmm_(self, batch1: 'Tensor', batch2: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  108. def addcdiv(self, tensor1: 'Tensor', tensor2: 'Tensor', *, value: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  109. def addcdiv_(self, tensor1: 'Tensor', tensor2: 'Tensor', *, value: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  110. def addcmul(self, tensor1: 'Tensor', tensor2: 'Tensor', *, value: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  111. def addcmul_(self, tensor1: 'Tensor', tensor2: 'Tensor', *, value: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  112. def addmm(self, mat1: 'Tensor', mat2: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  113. def addmm_(self, mat1: 'Tensor', mat2: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  114. def addmv(self, mat: 'Tensor', vec: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  115. def addmv_(self, mat: 'Tensor', vec: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  116. def addr(self, vec1: 'Tensor', vec2: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  117. def addr_(self, vec1: 'Tensor', vec2: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  118. @overload
  119. def all(self, *, out: Optional['Tensor']=None) -> Union[builtins.float, builtins.int]: ...
  120. @overload
  121. def all(self, dim: builtins.int, keepdim: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  122. def allclose(self, other: 'Tensor', rtol: builtins.float=1e-05, atol: builtins.float=1e-08, equal_nan: bool=False) -> bool: ...
  123. @overload
  124. def any(self, *, out: Optional['Tensor']=None) -> Union[builtins.float, builtins.int]: ...
  125. @overload
  126. def any(self, dim: builtins.int, keepdim: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  127. def apply_(self, callable: Callable) -> 'Tensor': ...
  128. def argmax(self, dim=None, keepdim=False): ...
  129. def argmin(self, dim=None, keepdim=False): ...
  130. def argsort(self, dim=None, descending=False): ...
  131. def asin(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  132. def asin_(self) -> 'Tensor': ...
  133. def atan(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  134. def atan2(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  135. def atan2_(self, other: 'Tensor') -> 'Tensor': ...
  136. def atan_(self) -> 'Tensor': ...
  137. def backward(self, gradient=None, retain_graph=None, create_graph=False): ...
  138. def baddbmm(self, batch1: 'Tensor', batch2: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  139. def baddbmm_(self, batch1: 'Tensor', batch2: 'Tensor', *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  140. @overload
  141. def bernoulli(self, *, generator: Generator=None, out: Optional['Tensor']=None) -> 'Tensor': ...
  142. @overload
  143. def bernoulli(self, p: builtins.float, *, generator: Generator=None, out: Optional['Tensor']=None) -> 'Tensor': ...
  144. @overload
  145. def bernoulli_(self, p: 'Tensor', *, generator: Generator=None) -> 'Tensor': ...
  146. @overload
  147. def bernoulli_(self, p: builtins.float=0.5, *, generator: Generator=None) -> 'Tensor': ...
  148. def bincount(self, weights: Optional['Tensor']=None, minlength: builtins.int=0) -> 'Tensor': ...
  149. def bmm(self, mat2: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  150. def btrifact(self, info=None, pivot=True): ...
  151. def btrifact_with_info(self, *, pivot: bool=True, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor', 'Tensor']: ...
  152. def btrisolve(self, LU_data: 'Tensor', LU_pivots: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  153. def byte(self) -> 'Tensor': ...
  154. def cauchy_(self, median: builtins.float=0, sigma: builtins.float=1, *, generator: Generator=None) -> 'Tensor': ...
  155. def ceil(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  156. def ceil_(self) -> 'Tensor': ...
  157. def char(self) -> 'Tensor': ...
  158. def chunk(self, chunks: builtins.int, dim: builtins.int=0) -> Union[Tuple['Tensor', ...],List['Tensor']]: ...
  159. def clamp(self, min: builtins.float=-math.inf, max: builtins.float =math.inf, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  160. def clamp_(self, min: builtins.float=-math.inf, max: builtins.float =math.inf) -> 'Tensor': ...
  161. def clone(self) -> 'Tensor': ...
  162. def contiguous(self) -> 'Tensor': ...
  163. def copy_(self, src: 'Tensor', non_blocking: bool=False) -> 'Tensor': ...
  164. def cos(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  165. def cos_(self) -> 'Tensor': ...
  166. def cosh(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  167. def cosh_(self) -> 'Tensor': ...
  168. def cpu(self) -> 'Tensor': ...
  169. def cross(self, other: 'Tensor', dim: builtins.int=-1, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  170. def cuda(self, device: Optional[device]=None, non_blocking: bool=False) -> 'Tensor': ...
  171. @overload
  172. def cumprod(self, dim: builtins.int, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  173. @overload
  174. def cumprod(self, dim: builtins.int, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  175. @overload
  176. def cumsum(self, dim: builtins.int, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  177. @overload
  178. def cumsum(self, dim: builtins.int, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  179. def data_ptr(self) -> builtins.int: ...
  180. def det(self) -> 'Tensor': ...
  181. def detach(self) -> 'Tensor': ...
  182. def detach_(self) -> 'Tensor': ...
  183. def diag(self, diagonal: builtins.int=0, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  184. def diagflat(self, offset: builtins.int=0) -> 'Tensor': ...
  185. def diagonal(self, offset: builtins.int=0, dim1: builtins.int=0, dim2: builtins.int=1) -> 'Tensor': ...
  186. def digamma(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  187. def digamma_(self) -> 'Tensor': ...
  188. def dim(self) -> builtins.int: ...
  189. def dist(self, other: 'Tensor', p: Union[builtins.float, builtins.int]=2) -> Union[builtins.float, builtins.int]: ...
  190. @overload
  191. def div(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  192. @overload
  193. def div(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  194. @overload
  195. def div_(self, other: 'Tensor') -> 'Tensor': ...
  196. @overload
  197. def div_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  198. def dot(self, tensor: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  199. def double(self) -> 'Tensor': ...
  200. def eig(self, eigenvectors: bool=False, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  201. def element_size(self) -> builtins.int: ...
  202. @overload
  203. def eq(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  204. @overload
  205. def eq(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  206. @overload
  207. def eq_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  208. @overload
  209. def eq_(self, other: 'Tensor') -> 'Tensor': ...
  210. def equal(self, other: 'Tensor') -> bool: ...
  211. def erf(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  212. def erf_(self) -> 'Tensor': ...
  213. def erfc(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  214. def erfc_(self) -> 'Tensor': ...
  215. def erfinv(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  216. def erfinv_(self) -> 'Tensor': ...
  217. def exp(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  218. def exp_(self) -> 'Tensor': ...
  219. @overload
  220. def expand(self, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, implicit: bool=False) -> 'Tensor': ...
  221. @overload
  222. def expand(self, *size: builtins.int, implicit: bool=False) -> 'Tensor': ...
  223. def expand_as(self, other: 'Tensor') -> 'Tensor': ...
  224. def expm1(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  225. def expm1_(self) -> 'Tensor': ...
  226. def exponential_(self, lambd: builtins.float=1, *, generator: Generator=None) -> 'Tensor': ...
  227. def fft(self, signal_ndim: builtins.int, normalized: bool=False) -> 'Tensor': ...
  228. @overload
  229. def fill_(self, value: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  230. @overload
  231. def fill_(self, value: 'Tensor') -> 'Tensor': ...
  232. def flatten(self, start_dim: builtins.int=0, end_dim: builtins.int=-1) -> 'Tensor': ...
  233. @overload
  234. def flip(self, dims: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> 'Tensor': ...
  235. @overload
  236. def flip(self, *dims: builtins.int) -> 'Tensor': ...
  237. def float(self) -> 'Tensor': ...
  238. def floor(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  239. def floor_(self) -> 'Tensor': ...
  240. @overload
  241. def fmod(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  242. @overload
  243. def fmod(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  244. @overload
  245. def fmod_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  246. @overload
  247. def fmod_(self, other: 'Tensor') -> 'Tensor': ...
  248. def frac(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  249. def frac_(self) -> 'Tensor': ...
  250. def gather(self, dim: builtins.int, index: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  251. @overload
  252. def ge(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  253. @overload
  254. def ge(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  255. @overload
  256. def ge_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  257. @overload
  258. def ge_(self, other: 'Tensor') -> 'Tensor': ...
  259. def gels(self, A: 'Tensor', *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  260. def geometric_(self, p: builtins.float, *, generator: Generator=None) -> 'Tensor': ...
  261. def geqrf(self, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  262. def ger(self, vec2: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  263. def gesv(self, A: 'Tensor', *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  264. def get_device(self) -> builtins.int: ...
  265. @overload
  266. def gt(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  267. @overload
  268. def gt(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  269. @overload
  270. def gt_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  271. @overload
  272. def gt_(self, other: 'Tensor') -> 'Tensor': ...
  273. def half(self) -> 'Tensor': ...
  274. def hardshrink(self, lambd: Union[builtins.float, builtins.int]=0.5) -> 'Tensor': ...
  275. def histc(self, bins: builtins.int=100, min: Union[builtins.float, builtins.int]=0, max: Union[builtins.float, builtins.int]=0, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  276. def ifft(self, signal_ndim: builtins.int, normalized: bool=False) -> 'Tensor': ...
  277. def index_add(self, dim, index, tensor): ...
  278. def index_add_(self, dim: builtins.int, index: 'Tensor', source: 'Tensor') -> 'Tensor': ...
  279. def index_copy(self, dim, index, tensor): ...
  280. def index_copy_(self, dim: builtins.int, index: 'Tensor', source: 'Tensor') -> 'Tensor': ...
  281. def index_fill(self, dim, index, value): ...
  282. @overload
  283. def index_fill_(self, dim: builtins.int, index: 'Tensor', value: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  284. @overload
  285. def index_fill_(self, dim: builtins.int, index: 'Tensor', value: 'Tensor') -> 'Tensor': ...
  286. def index_put_(self, indices: Union[Tuple['Tensor', ...],List['Tensor']], values: 'Tensor') -> 'Tensor': ...
  287. def index_select(self, dim: builtins.int, index: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  288. def int(self) -> 'Tensor': ...
  289. def inverse(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  290. def irfft(self, signal_ndim: builtins.int, normalized: bool=False, onesided: bool=True, signal_sizes: Union[Tuple[builtins.int, ...], List[builtins.int], Size]=()) -> 'Tensor': ...
  291. def is_contiguous(self) -> bool: ...
  292. def is_pinned(self): ...
  293. def is_set_to(self, tensor: 'Tensor') -> bool: ...
  294. def is_shared(self): ...
  295. def item(self) -> Union[builtins.float, builtins.int]: ...
  296. def kthvalue(self, k: builtins.int, dim: builtins.int=-1, keepdim: bool=False, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  297. @overload
  298. def le(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  299. @overload
  300. def le(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  301. @overload
  302. def le_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  303. @overload
  304. def le_(self, other: 'Tensor') -> 'Tensor': ...
  305. def lerp(self, end: 'Tensor', weight: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  306. def lerp_(self, end: 'Tensor', weight: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  307. def log(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  308. def log10(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  309. def log10_(self) -> 'Tensor': ...
  310. def log1p(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  311. def log1p_(self) -> 'Tensor': ...
  312. def log2(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  313. def log2_(self) -> 'Tensor': ...
  314. def log_(self) -> 'Tensor': ...
  315. def log_normal_(self, mean: builtins.float=1, std: builtins.float=2, *, generator: Generator=None) -> 'Tensor': ...
  316. def logdet(self) -> 'Tensor': ...
  317. def logsumexp(self, dim: builtins.int, keepdim: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  318. def long(self) -> 'Tensor': ...
  319. @overload
  320. def lt(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  321. @overload
  322. def lt(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  323. @overload
  324. def lt_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  325. @overload
  326. def lt_(self, other: 'Tensor') -> 'Tensor': ...
  327. def map_(tensor: 'Tensor', callable: Callable) -> 'Tensor': ...
  328. def masked_fill(self, mask, value): ...
  329. @overload
  330. def masked_fill_(self, mask: 'Tensor', value: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  331. @overload
  332. def masked_fill_(self, mask: 'Tensor', value: 'Tensor') -> 'Tensor': ...
  333. def masked_scatter(self, mask, tensor): ...
  334. def masked_scatter_(self, mask: 'Tensor', source: 'Tensor') -> 'Tensor': ...
  335. def masked_select(self, mask: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  336. def matmul(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  337. def matrix_power(self, n: builtins.int) -> 'Tensor': ...
  338. @overload
  339. def max(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  340. @overload
  341. def max(self, *, out: Optional['Tensor']=None) -> Union[builtins.float, builtins.int]: ...
  342. @overload
  343. def max(self, dim: builtins.int, keepdim: bool=False, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  344. @overload
  345. def mean(self, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  346. @overload
  347. def mean(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  348. @overload
  349. def mean(self, dim: builtins.int, keepdim: bool, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  350. @overload
  351. def mean(self, dim: builtins.int, keepdim: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  352. @overload
  353. def mean(self, dim: builtins.int, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  354. @overload
  355. def median(self, *, out: Optional['Tensor']=None) -> Union[builtins.float, builtins.int]: ...
  356. @overload
  357. def median(self, dim: builtins.int, keepdim: bool=False, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  358. @overload
  359. def min(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  360. @overload
  361. def min(self, *, out: Optional['Tensor']=None) -> Union[builtins.float, builtins.int]: ...
  362. @overload
  363. def min(self, dim: builtins.int, keepdim: bool=False, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  364. def mm(self, mat2: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  365. def mode(self, dim: builtins.int=-1, keepdim: bool=False, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  366. @overload
  367. def mul(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  368. @overload
  369. def mul(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  370. @overload
  371. def mul_(self, other: 'Tensor') -> 'Tensor': ...
  372. @overload
  373. def mul_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  374. def multinomial(self, num_samples: builtins.int, replacement: bool=False, *, generator: Generator=None, out: Optional['Tensor']=None) -> 'Tensor': ...
  375. def mv(self, vec: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  376. def mvlgamma(self, p: builtins.int) -> 'Tensor': ...
  377. def mvlgamma_(self, p: builtins.int) -> 'Tensor': ...
  378. def narrow(self, dim: builtins.int, start: builtins.int, length: builtins.int) -> 'Tensor': ...
  379. def narrow_copy(self, dim: builtins.int, start: builtins.int, length: builtins.int) -> 'Tensor': ...
  380. def ndimension(self) -> builtins.int: ...
  381. @overload
  382. def ne(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  383. @overload
  384. def ne(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  385. @overload
  386. def ne_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  387. @overload
  388. def ne_(self, other: 'Tensor') -> 'Tensor': ...
  389. def neg(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  390. def neg_(self) -> 'Tensor': ...
  391. def nelement(self) -> builtins.int: ...
  392. def new_empty(self, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], dtype: Optional[_dtype]=None, device: Union[device, str, None]=None, requires_grad: bool=False) -> 'Tensor': ...
  393. def new_full(self, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], value: Union[builtins.float, builtins.int], dtype: Optional[_dtype]=None, device: Union[device, str, None]=None, requires_grad: bool=False) -> 'Tensor': ...
  394. def new_ones(self, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], dtype: Optional[_dtype]=None, device: Union[device, str, None]=None, requires_grad: bool=False) -> 'Tensor': ...
  395. def new_tensor(self, data: Any, dtype: Optional[_dtype]=None, device: Union[device, str, None]=None, requires_grad: bool=False) -> 'Tensor': ...
  396. def new_zeros(self, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], dtype: Optional[_dtype]=None, device: Union[device, str, None]=None, requires_grad: bool=False) -> 'Tensor': ...
  397. def nonzero(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  398. def norm(self, p='fro', dim=None, keepdim=False): ...
  399. def normal_(self, mean: builtins.float=0, std: builtins.float=1, *, generator: Generator=None) -> 'Tensor': ...
  400. def numel(self) -> builtins.int: ...
  401. def numpy(self) -> Any: ...
  402. def orgqr(self, input2: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  403. def ormqr(self, input2: 'Tensor', input3: 'Tensor', left: bool=True, transpose: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  404. @overload
  405. def permute(self, dims: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> 'Tensor': ...
  406. @overload
  407. def permute(self, *dims: builtins.int) -> 'Tensor': ...
  408. def pinverse(self, rcond: builtins.float=1e-15) -> 'Tensor': ...
  409. def potrf(self, upper: bool=True, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  410. def potri(self, upper: bool=True, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  411. def potrs(self, input2: 'Tensor', upper: bool=True, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  412. @overload
  413. def pow(self, exponent: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  414. @overload
  415. def pow(self, exponent: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  416. @overload
  417. def pow_(self, exponent: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  418. @overload
  419. def pow_(self, exponent: 'Tensor') -> 'Tensor': ...
  420. @overload
  421. def prod(self, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  422. @overload
  423. def prod(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  424. @overload
  425. def prod(self, dim: builtins.int, keepdim: bool, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  426. @overload
  427. def prod(self, dim: builtins.int, keepdim: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  428. @overload
  429. def prod(self, dim: builtins.int, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  430. def pstrf(self, upper: bool=True, tol: Union[builtins.float, builtins.int]=-1, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  431. def put_(self, index: 'Tensor', source: 'Tensor', accumulate: bool=False) -> 'Tensor': ...
  432. def qr(self, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  433. @overload
  434. def random_(self, from_: builtins.int, to: builtins.int, *, generator: Generator=None) -> 'Tensor': ...
  435. @overload
  436. def random_(self, to: builtins.int, *, generator: Generator=None) -> 'Tensor': ...
  437. @overload
  438. def random_(self, *, generator: Generator=None) -> 'Tensor': ...
  439. def reciprocal(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  440. def reciprocal_(self) -> 'Tensor': ...
  441. def register_hook(self, hook): ...
  442. @overload
  443. def remainder(self, other: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  444. @overload
  445. def remainder(self, other: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  446. @overload
  447. def remainder_(self, other: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  448. @overload
  449. def remainder_(self, other: 'Tensor') -> 'Tensor': ...
  450. def renorm(self, p: Union[builtins.float, builtins.int], dim: builtins.int, maxnorm: Union[builtins.float, builtins.int], *, out: Optional['Tensor']=None) -> 'Tensor': ...
  451. def renorm_(self, p: Union[builtins.float, builtins.int], dim: builtins.int, maxnorm: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  452. @overload
  453. def repeat(self, repeats: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> 'Tensor': ...
  454. @overload
  455. def repeat(self, *repeats: builtins.int) -> 'Tensor': ...
  456. def requires_grad_(self, mode: bool=True) -> 'Tensor': ...
  457. @overload
  458. def reshape(self, shape: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> 'Tensor': ...
  459. @overload
  460. def reshape(self, *shape: builtins.int) -> 'Tensor': ...
  461. def reshape_as(self, other: 'Tensor') -> 'Tensor': ...
  462. @overload
  463. def resize_(self, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> 'Tensor': ...
  464. @overload
  465. def resize_(self, *size: builtins.int) -> 'Tensor': ...
  466. def resize_as_(self, the_template: 'Tensor') -> 'Tensor': ...
  467. def retain_grad(self): ...
  468. def rfft(self, signal_ndim: builtins.int, normalized: bool=False, onesided: bool=True) -> 'Tensor': ...
  469. def rot90(self, k: builtins.int=1, dims: Union[Tuple[builtins.int, ...], List[builtins.int], Size]=(0,1)) -> 'Tensor': ...
  470. def round(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  471. def round_(self) -> 'Tensor': ...
  472. def rsqrt(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  473. def rsqrt_(self) -> 'Tensor': ...
  474. def scatter(self, dim, index, source): ...
  475. @overload
  476. def scatter_(self, dim: builtins.int, index: 'Tensor', src: 'Tensor') -> 'Tensor': ...
  477. @overload
  478. def scatter_(self, dim: builtins.int, index: 'Tensor', value: Union[builtins.float, builtins.int]) -> 'Tensor': ...
  479. def scatter_add(self, dim, index, source): ...
  480. def scatter_add_(self, dim: builtins.int, index: 'Tensor', src: 'Tensor') -> 'Tensor': ...
  481. def select(self, dim: builtins.int, index: builtins.int) -> 'Tensor': ...
  482. @overload
  483. def set_(self, source: Storage) -> 'Tensor': ...
  484. @overload
  485. def set_(self, source: Storage, storage_offset: builtins.int, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], stride: Union[Tuple[builtins.int, ...], List[builtins.int], Size]=()) -> 'Tensor': ...
  486. @overload
  487. def set_(self, source: 'Tensor') -> 'Tensor': ...
  488. @overload
  489. def set_(self) -> 'Tensor': ...
  490. def share_memory_(self): ...
  491. def short(self) -> 'Tensor': ...
  492. def sigmoid(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  493. def sigmoid_(self) -> 'Tensor': ...
  494. def sign(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  495. def sign_(self) -> 'Tensor': ...
  496. def sin(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  497. def sin_(self) -> 'Tensor': ...
  498. def sinh(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  499. def sinh_(self) -> 'Tensor': ...
  500. @overload
  501. def size(self) -> Size: ...
  502. @overload
  503. def size(self, dim: builtins.int) -> builtins.int: ...
  504. def slogdet(self) -> Tuple['Tensor', 'Tensor']: ...
  505. def sort(self, dim: builtins.int=-1, descending: bool=False, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  506. def split(self, split_size, dim=0): ...
  507. def sqrt(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  508. def sqrt_(self) -> 'Tensor': ...
  509. @overload
  510. def squeeze(self) -> 'Tensor': ...
  511. @overload
  512. def squeeze(self, dim: builtins.int) -> 'Tensor': ...
  513. @overload
  514. def squeeze_(self) -> 'Tensor': ...
  515. @overload
  516. def squeeze_(self, dim: builtins.int) -> 'Tensor': ...
  517. @overload
  518. def std(self, unbiased: bool=True, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  519. @overload
  520. def std(self, dim: builtins.int, unbiased: bool=True, keepdim: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  521. def stft(self, n_fft, hop_length=None, win_length=None, window=None, center=True, pad_mode='reflect', normalized=False, onesided=True): ...
  522. def storage(self) -> Storage: ...
  523. def storage_offset(self) -> builtins.int: ...
  524. @overload
  525. def stride(self) -> Tuple[builtins.int]: ...
  526. @overload
  527. def stride(self, dim: builtins.int) -> builtins.int: ...
  528. @overload
  529. def sub(self, other: 'Tensor', *, alpha: Union[builtins.float, builtins.int]=1, out: Optional['Tensor']=None) -> 'Tensor': ...
  530. @overload
  531. def sub(self, other: Union[builtins.float, builtins.int], alpha: Union[builtins.float, builtins.int]=1, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  532. @overload
  533. def sub_(self, other: 'Tensor', *, alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  534. @overload
  535. def sub_(self, other: Union[builtins.float, builtins.int], alpha: Union[builtins.float, builtins.int]=1) -> 'Tensor': ...
  536. @overload
  537. def sum(self, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  538. @overload
  539. def sum(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  540. @overload
  541. def sum(self, dim: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size], keepdim: bool, *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  542. @overload
  543. def sum(self, dim: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size], keepdim: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  544. @overload
  545. def sum(self, dim: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size], *, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  546. @overload
  547. def sum(self, *dim: builtins.int, dtype: _dtype, out: Optional['Tensor']=None) -> 'Tensor': ...
  548. def svd(self, some: bool=True, compute_uv: bool=True, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor', 'Tensor']: ...
  549. def symeig(self, eigenvectors: bool=False, upper: bool=True, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  550. def t(self) -> 'Tensor': ...
  551. def t_(self) -> 'Tensor': ...
  552. def take(self, index: 'Tensor', *, out: Optional['Tensor']=None) -> 'Tensor': ...
  553. def tan_(self) -> 'Tensor': ...
  554. def tanh(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  555. def tanh_(self) -> 'Tensor': ...
  556. @overload
  557. def to(self, device: Union[device, str, None], dtype: _dtype, non_blocking: bool=False, copy: bool=False) -> 'Tensor': ...
  558. @overload
  559. def to(self, dtype: _dtype, non_blocking: bool=False, copy: bool=False) -> 'Tensor': ...
  560. @overload
  561. def to(self, device: Union[device, str, None], non_blocking: bool=False, copy: bool=False) -> 'Tensor': ...
  562. @overload
  563. def to(self, other: 'Tensor', non_blocking: bool=False, copy: bool=False) -> 'Tensor': ...
  564. def tolist(self) -> List: ...
  565. def topk(self, k: builtins.int, dim: builtins.int=-1, largest: bool=True, sorted: bool=True, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  566. def trace(self) -> Union[builtins.float, builtins.int]: ...
  567. def transpose(self, dim0: builtins.int, dim1: builtins.int) -> 'Tensor': ...
  568. def transpose_(self, dim0: builtins.int, dim1: builtins.int) -> 'Tensor': ...
  569. def tril(self, diagonal: builtins.int=0, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  570. def tril_(self, diagonal: builtins.int=0) -> 'Tensor': ...
  571. def triu(self, diagonal: builtins.int=0, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  572. def triu_(self, diagonal: builtins.int=0) -> 'Tensor': ...
  573. def trtrs(self, A: 'Tensor', upper: bool=True, transpose: bool=False, unitriangular: bool=False, *, out: Optional['Tensor']=None) -> Tuple['Tensor', 'Tensor']: ...
  574. def trunc(self, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  575. def trunc_(self) -> 'Tensor': ...
  576. def type(self, dtype: Union[None, str, _dtype]=None, non_blocking: bool=False) -> Union[str, 'Tensor']: ...
  577. def type_as(self, other: 'Tensor') -> 'Tensor': ...
  578. def unbind(self, dim: builtins.int=0) -> Union[Tuple['Tensor', ...],List['Tensor']]: ...
  579. def unfold(self, dimension: builtins.int, size: builtins.int, step: builtins.int) -> 'Tensor': ...
  580. def uniform_(self, from_: builtins.float=0, to: builtins.float=1, *, generator: Generator=None) -> 'Tensor': ...
  581. def unique(self, sorted=False, return_inverse=False, dim=None): ...
  582. def unsqueeze(self, dim: builtins.int) -> 'Tensor': ...
  583. def unsqueeze_(self, dim: builtins.int) -> 'Tensor': ...
  584. @overload
  585. def var(self, unbiased: bool=True, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  586. @overload
  587. def var(self, dim: builtins.int, unbiased: bool=True, keepdim: bool=False, *, out: Optional['Tensor']=None) -> 'Tensor': ...
  588. @overload
  589. def view(self, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> 'Tensor': ...
  590. @overload
  591. def view(self, *size: builtins.int) -> 'Tensor': ...
  592. def view_as(self, other: 'Tensor') -> 'Tensor': ...
  593. def where(self, condition: 'Tensor', other: 'Tensor') -> 'Tensor': ...
  594. def zero_(self) -> 'Tensor': ...
  595. def abs(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  596. def acos(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  597. def adaptive_avg_pool1d(self: Tensor, output_size: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]) -> Tensor: ...
  598. @overload
  599. def add(self: Tensor, other: Tensor, *, alpha: Union[builtins.float, builtins.int]=1, out: Optional[Tensor]=None) -> Tensor: ...
  600. @overload
  601. def add(self: Tensor, other: Union[builtins.float, builtins.int], alpha: Union[builtins.float, builtins.int]=1, *, out: Optional[Tensor]=None) -> Tensor: ...
  602. @overload
  603. def add(self: Tensor, alpha: Union[builtins.float, builtins.int], other: Tensor) -> Tensor: ...
  604. @overload
  605. def add(self: Tensor, alpha: Union[builtins.float, builtins.int], other: Tensor, *, out: Tensor) -> Tensor: ...
  606. @overload
  607. def addbmm(self: Tensor, batch1: Tensor, batch2: Tensor, *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional[Tensor]=None) -> Tensor: ...
  608. @overload
  609. def addbmm(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], batch1: Tensor, batch2: Tensor) -> Tensor: ...
  610. @overload
  611. def addbmm(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], batch1: Tensor, batch2: Tensor, *, out: Tensor) -> Tensor: ...
  612. @overload
  613. def addbmm(beta: Union[builtins.float, builtins.int], self: Tensor, batch1: Tensor, batch2: Tensor) -> Tensor: ...
  614. @overload
  615. def addbmm(beta: Union[builtins.float, builtins.int], self: Tensor, batch1: Tensor, batch2: Tensor, *, out: Tensor) -> Tensor: ...
  616. @overload
  617. def addcdiv(self: Tensor, tensor1: Tensor, tensor2: Tensor, *, value: Union[builtins.float, builtins.int]=1, out: Optional[Tensor]=None) -> Tensor: ...
  618. @overload
  619. def addcdiv(self: Tensor, value: Union[builtins.float, builtins.int], tensor1: Tensor, tensor2: Tensor) -> Tensor: ...
  620. @overload
  621. def addcdiv(self: Tensor, value: Union[builtins.float, builtins.int], tensor1: Tensor, tensor2: Tensor, *, out: Tensor) -> Tensor: ...
  622. @overload
  623. def addcmul(self: Tensor, tensor1: Tensor, tensor2: Tensor, *, value: Union[builtins.float, builtins.int]=1, out: Optional[Tensor]=None) -> Tensor: ...
  624. @overload
  625. def addcmul(self: Tensor, value: Union[builtins.float, builtins.int], tensor1: Tensor, tensor2: Tensor) -> Tensor: ...
  626. @overload
  627. def addcmul(self: Tensor, value: Union[builtins.float, builtins.int], tensor1: Tensor, tensor2: Tensor, *, out: Tensor) -> Tensor: ...
  628. @overload
  629. def addmm(self: Tensor, mat1: Tensor, mat2: Tensor, *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional[Tensor]=None) -> Tensor: ...
  630. @overload
  631. def addmm(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], mat1: Tensor, mat2: Tensor) -> Tensor: ...
  632. @overload
  633. def addmm(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], mat1: Tensor, mat2: Tensor, *, out: Tensor) -> Tensor: ...
  634. @overload
  635. def addmm(beta: Union[builtins.float, builtins.int], self: Tensor, mat1: Tensor, mat2: Tensor) -> Tensor: ...
  636. @overload
  637. def addmm(beta: Union[builtins.float, builtins.int], self: Tensor, mat1: Tensor, mat2: Tensor, *, out: Tensor) -> Tensor: ...
  638. @overload
  639. def addmv(self: Tensor, mat: Tensor, vec: Tensor, *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional[Tensor]=None) -> Tensor: ...
  640. @overload
  641. def addmv(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], mat: Tensor, vec: Tensor) -> Tensor: ...
  642. @overload
  643. def addmv(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], mat: Tensor, vec: Tensor, *, out: Tensor) -> Tensor: ...
  644. @overload
  645. def addmv(beta: Union[builtins.float, builtins.int], self: Tensor, mat: Tensor, vec: Tensor) -> Tensor: ...
  646. @overload
  647. def addmv(beta: Union[builtins.float, builtins.int], self: Tensor, mat: Tensor, vec: Tensor, *, out: Tensor) -> Tensor: ...
  648. @overload
  649. def addr(self: Tensor, vec1: Tensor, vec2: Tensor, *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional[Tensor]=None) -> Tensor: ...
  650. @overload
  651. def addr(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], vec1: Tensor, vec2: Tensor) -> Tensor: ...
  652. @overload
  653. def addr(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], vec1: Tensor, vec2: Tensor, *, out: Tensor) -> Tensor: ...
  654. @overload
  655. def addr(beta: Union[builtins.float, builtins.int], self: Tensor, vec1: Tensor, vec2: Tensor) -> Tensor: ...
  656. @overload
  657. def addr(beta: Union[builtins.float, builtins.int], self: Tensor, vec1: Tensor, vec2: Tensor, *, out: Tensor) -> Tensor: ...
  658. def allclose(self: Tensor, other: Tensor, rtol: builtins.float=1e-05, atol: builtins.float=1e-08, equal_nan: bool=False) -> bool: ...
  659. @overload
  660. def arange(start: Union[builtins.float, builtins.int], end: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  661. @overload
  662. def arange(start: Union[builtins.float, builtins.int], end: Union[builtins.float, builtins.int], step: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  663. @overload
  664. def arange(end: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  665. def argmax(input, dim=None, keepdim=False): ...
  666. def argmin(input, dim=None, keepdim=False): ...
  667. def argsort(input, dim=None, descending=False): ...
  668. def as_tensor(data: Any, dtype: _dtype=None, device: Optional[device]=None) -> Tensor: ...
  669. def asin(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  670. def atan(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  671. def atan2(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  672. def avg_pool1d(self: Tensor, kernel_size: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size], stride: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=(), padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, ceil_mode: bool=False, count_include_pad: bool=True) -> Tensor: ...
  673. @overload
  674. def baddbmm(self: Tensor, batch1: Tensor, batch2: Tensor, *, beta: Union[builtins.float, builtins.int]=1, alpha: Union[builtins.float, builtins.int]=1, out: Optional[Tensor]=None) -> Tensor: ...
  675. @overload
  676. def baddbmm(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], batch1: Tensor, batch2: Tensor) -> Tensor: ...
  677. @overload
  678. def baddbmm(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], batch1: Tensor, batch2: Tensor, *, out: Tensor) -> Tensor: ...
  679. @overload
  680. def baddbmm(beta: Union[builtins.float, builtins.int], self: Tensor, batch1: Tensor, batch2: Tensor) -> Tensor: ...
  681. @overload
  682. def baddbmm(beta: Union[builtins.float, builtins.int], self: Tensor, batch1: Tensor, batch2: Tensor, *, out: Tensor) -> Tensor: ...
  683. @overload
  684. def bartlett_window(window_length: builtins.int, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  685. @overload
  686. def bartlett_window(window_length: builtins.int, periodic: bool, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  687. @overload
  688. def bernoulli(self: Tensor, *, generator: Generator=None, out: Optional[Tensor]=None) -> Tensor: ...
  689. @overload
  690. def bernoulli(self: Tensor, p: builtins.float, *, generator: Generator=None, out: Optional[Tensor]=None) -> Tensor: ...
  691. def bincount(self: Tensor, weights: Optional[Tensor]=None, minlength: builtins.int=0) -> Tensor: ...
  692. @overload
  693. def blackman_window(window_length: builtins.int, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  694. @overload
  695. def blackman_window(window_length: builtins.int, periodic: bool, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  696. def bmm(self: Tensor, mat2: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  697. def broadcast_tensors(*tensors:Tensor) -> List[Tensor]: ...
  698. def btrifact(A:Tensor, info:Union[Tensor, None]=None, pivot:bool=True) -> Tuple[Tensor, Tensor]: ...
  699. def btrifact_with_info(self: Tensor, *, pivot: bool=True, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor, Tensor]: ...
  700. def btrisolve(self: Tensor, LU_data: Tensor, LU_pivots: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  701. def btriunpack(LU_data:Tensor, LU_pivots:Tensor, unpack_data:bool=True, unpack_pivots:bool=True) -> Tuple[Tensor, Tensor, Tensor]: ...
  702. def cat(tensors: Union[Tuple[Tensor, ...],List[Tensor]], dim: builtins.int=0, *, out: Optional[Tensor]=None) -> Tensor: ...
  703. def ceil(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  704. def celu_(self: Tensor, alpha: Union[builtins.float, builtins.int]=1.0) -> Tensor: ...
  705. def chain_matmul(*matrices): ...
  706. def chunk(self: Tensor, chunks: builtins.int, dim: builtins.int=0) -> Union[Tuple[Tensor, ...],List[Tensor]]: ...
  707. def clamp(self, min: builtins.float=-math.inf, max: builtins.float=math.inf, *, out: Optional[Tensor]=None) -> Tensor: ...
  708. def compiled_with_cxx11_abi(): ...
  709. def conv1d(input: Tensor, weight: Tensor, bias: Tensor=None, stride: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, dilation: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, groups: builtins.int=1) -> Tensor: ...
  710. def conv2d(input: Tensor, weight: Tensor, bias: Tensor=None, stride: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, dilation: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, groups: builtins.int=1) -> Tensor: ...
  711. def conv3d(input: Tensor, weight: Tensor, bias: Tensor=None, stride: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, dilation: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, groups: builtins.int=1) -> Tensor: ...
  712. def conv_tbc(self: Tensor, weight: Tensor, bias: Tensor, pad: builtins.int=0) -> Tensor: ...
  713. def conv_transpose1d(input: Tensor, weight: Tensor, bias: Tensor=None, stride: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, output_padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, groups: builtins.int=1, dilation: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1) -> Tensor: ...
  714. def conv_transpose2d(input: Tensor, weight: Tensor, bias: Tensor=None, stride: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, output_padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, groups: builtins.int=1, dilation: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1) -> Tensor: ...
  715. def conv_transpose3d(input: Tensor, weight: Tensor, bias: Tensor=None, stride: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1, padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, output_padding: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=0, groups: builtins.int=1, dilation: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size]=1) -> Tensor: ...
  716. def cos(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  717. def cosh(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  718. def cross(self: Tensor, other: Tensor, dim: builtins.int=-1, *, out: Optional[Tensor]=None) -> Tensor: ...
  719. @overload
  720. def cumprod(self: Tensor, dim: builtins.int, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  721. @overload
  722. def cumprod(self: Tensor, dim: builtins.int, *, out: Optional[Tensor]=None) -> Tensor: ...
  723. @overload
  724. def cumsum(self: Tensor, dim: builtins.int, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  725. @overload
  726. def cumsum(self: Tensor, dim: builtins.int, *, out: Optional[Tensor]=None) -> Tensor: ...
  727. def det(self: Tensor) -> Tensor: ...
  728. def diag(self: Tensor, diagonal: builtins.int=0, *, out: Optional[Tensor]=None) -> Tensor: ...
  729. def diagflat(self: Tensor, offset: builtins.int=0) -> Tensor: ...
  730. def diagonal(self: Tensor, offset: builtins.int=0, dim1: builtins.int=0, dim2: builtins.int=1) -> Tensor: ...
  731. def digamma(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  732. def dist(self: Tensor, other: Tensor, p: Union[builtins.float, builtins.int]=2) -> Union[builtins.float, builtins.int]: ...
  733. @overload
  734. def div(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  735. @overload
  736. def div(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  737. def dot(self: Tensor, tensor: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  738. def eig(self: Tensor, eigenvectors: bool=False, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  739. def einsum(equation:str, *operands:Tensor) -> Tensor: ...
  740. @overload
  741. def empty(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  742. @overload
  743. def empty(*size: builtins.int, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  744. @overload
  745. def empty_like(self: Tensor) -> Tensor: ...
  746. @overload
  747. def empty_like(self: Tensor, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  748. @overload
  749. def eq(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  750. @overload
  751. def eq(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  752. def equal(self: Tensor, other: Tensor) -> bool: ...
  753. def erf(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  754. def erfc(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  755. def erfinv(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  756. def exp(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  757. def expm1(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  758. @overload
  759. def eye(n: builtins.int, *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  760. @overload
  761. def eye(n: builtins.int, m: builtins.int, *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  762. def fft(self: Tensor, signal_ndim: builtins.int, normalized: bool=False) -> Tensor: ...
  763. def flatten(self: Tensor, start_dim: builtins.int=0, end_dim: builtins.int=-1) -> Tensor: ...
  764. def flip(self: Tensor, dims: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> Tensor: ...
  765. def floor(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  766. @overload
  767. def fmod(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  768. @overload
  769. def fmod(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  770. def frac(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  771. def from_numpy(ndarray) -> Tensor: ...
  772. def full(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], fill_value: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  773. @overload
  774. def full_like(self: Tensor, fill_value: Union[builtins.float, builtins.int]) -> Tensor: ...
  775. @overload
  776. def full_like(self: Tensor, fill_value: Union[builtins.float, builtins.int], *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  777. def gather(self: Tensor, dim: builtins.int, index: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  778. @overload
  779. def ge(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  780. @overload
  781. def ge(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  782. def gels(self: Tensor, A: Tensor, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  783. def geqrf(self: Tensor, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  784. def ger(self: Tensor, vec2: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  785. def gesv(self: Tensor, A: Tensor, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  786. def get_default_dtype() -> _dtype: ...
  787. def get_num_threads() -> builtins.int: ...
  788. def get_rng_state(): ...
  789. @overload
  790. def gt(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  791. @overload
  792. def gt(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  793. @overload
  794. def hamming_window(window_length: builtins.int, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  795. @overload
  796. def hamming_window(window_length: builtins.int, periodic: bool, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  797. @overload
  798. def hamming_window(window_length: builtins.int, periodic: bool, alpha: builtins.float, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  799. @overload
  800. def hamming_window(window_length: builtins.int, periodic: bool, alpha: builtins.float, beta: builtins.float, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  801. @overload
  802. def hann_window(window_length: builtins.int, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  803. @overload
  804. def hann_window(window_length: builtins.int, periodic: bool, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  805. def histc(self: Tensor, bins: builtins.int=100, min: Union[builtins.float, builtins.int]=0, max: Union[builtins.float, builtins.int]=0, *, out: Optional[Tensor]=None) -> Tensor: ...
  806. def ifft(self: Tensor, signal_ndim: builtins.int, normalized: bool=False) -> Tensor: ...
  807. def index_select(self: Tensor, dim: builtins.int, index: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  808. def initial_seed(): ...
  809. def inverse(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  810. def irfft(self: Tensor, signal_ndim: builtins.int, normalized: bool=False, onesided: bool=True, signal_sizes: Union[Tuple[builtins.int, ...], List[builtins.int], Size]=()) -> Tensor: ...
  811. def is_storage(obj): ...
  812. def is_tensor(obj): ...
  813. def isfinite(tensor:Tensor) -> Tensor: ...
  814. def isinf(tensor:Tensor) -> Tensor: ...
  815. def isnan(tensor:Tensor) -> Tensor: ...
  816. def kthvalue(self: Tensor, k: builtins.int, dim: builtins.int=-1, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  817. @overload
  818. def le(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  819. @overload
  820. def le(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  821. def lerp(self: Tensor, end: Tensor, weight: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  822. @overload
  823. def linspace(start: Union[builtins.float, builtins.int], end: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  824. @overload
  825. def linspace(start: Union[builtins.float, builtins.int], end: Union[builtins.float, builtins.int], steps: builtins.int, *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  826. def load(f, map_location=None, pickle_module=pickle): ...
  827. def log(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  828. def log10(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  829. def log1p(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  830. def log2(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  831. def logdet(self: Tensor) -> Tensor: ...
  832. @overload
  833. def logspace(start: Union[builtins.float, builtins.int], end: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  834. @overload
  835. def logspace(start: Union[builtins.float, builtins.int], end: Union[builtins.float, builtins.int], steps: builtins.int, *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  836. def logsumexp(self: Tensor, dim: builtins.int, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tensor: ...
  837. @overload
  838. def lt(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  839. @overload
  840. def lt(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  841. def manual_seed(seed): ...
  842. def masked_select(self: Tensor, mask: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  843. def matmul(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  844. def matrix_power(self: Tensor, n: builtins.int) -> Tensor: ...
  845. @overload
  846. def matrix_rank(self: Tensor, tol: builtins.float, symmetric: bool=False) -> Tensor: ...
  847. @overload
  848. def matrix_rank(self: Tensor, symmetric: bool=False) -> Tensor: ...
  849. @overload
  850. def max(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  851. @overload
  852. def max(self: Tensor, *, out: Optional[Tensor]=None) -> Union[builtins.float, builtins.int]: ...
  853. @overload
  854. def max(self: Tensor, dim: builtins.int, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  855. @overload
  856. def mean(self: Tensor, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  857. @overload
  858. def mean(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  859. @overload
  860. def mean(self: Tensor, dim: builtins.int, keepdim: bool, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  861. @overload
  862. def mean(self: Tensor, dim: builtins.int, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tensor: ...
  863. @overload
  864. def mean(self: Tensor, dim: builtins.int, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  865. @overload
  866. def median(self: Tensor, *, out: Optional[Tensor]=None) -> Union[builtins.float, builtins.int]: ...
  867. @overload
  868. def median(self: Tensor, dim: builtins.int, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  869. def meshgrid(*tensors, **kwargs): ...
  870. @overload
  871. def min(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  872. @overload
  873. def min(self: Tensor, *, out: Optional[Tensor]=None) -> Union[builtins.float, builtins.int]: ...
  874. @overload
  875. def min(self: Tensor, dim: builtins.int, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  876. def mm(self: Tensor, mat2: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  877. def mode(self: Tensor, dim: builtins.int=-1, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  878. @overload
  879. def mul(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  880. @overload
  881. def mul(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  882. def multinomial(self: Tensor, num_samples: builtins.int, replacement: bool=False, *, generator: Generator=None, out: Optional[Tensor]=None) -> Tensor: ...
  883. def mv(self: Tensor, vec: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  884. def mvlgamma(self: Tensor, p: builtins.int) -> Tensor: ...
  885. def narrow(self: Tensor, dim: builtins.int, start: builtins.int, length: builtins.int) -> Tensor: ...
  886. @overload
  887. def ne(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  888. @overload
  889. def ne(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  890. def neg(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  891. def nonzero(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  892. def norm(input, p='fro', dim=None, keepdim=False, out=None): ...
  893. @overload
  894. def normal(mean: Tensor, std: builtins.float=1, *, generator: Generator=None, out: Optional[Tensor]=None) -> Tensor: ...
  895. @overload
  896. def normal(mean: builtins.float, std: Tensor, *, generator: Generator=None, out: Optional[Tensor]=None) -> Tensor: ...
  897. @overload
  898. def normal(mean: Tensor, std: Tensor, *, generator: Generator=None, out: Optional[Tensor]=None) -> Tensor: ...
  899. def numel(self: Tensor) -> builtins.int: ...
  900. @overload
  901. def ones(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  902. @overload
  903. def ones(*size: builtins.int, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  904. @overload
  905. def ones_like(self: Tensor) -> Tensor: ...
  906. @overload
  907. def ones_like(self: Tensor, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  908. def orgqr(self: Tensor, input2: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  909. def ormqr(self: Tensor, input2: Tensor, input3: Tensor, left: bool=True, transpose: bool=False, *, out: Optional[Tensor]=None) -> Tensor: ...
  910. def pdist(self: Tensor, p: builtins.float=2) -> Tensor: ...
  911. def pinverse(self: Tensor, rcond: builtins.float=1e-15) -> Tensor: ...
  912. def pixel_shuffle(self: Tensor, upscale_factor: builtins.int) -> Tensor: ...
  913. def potrf(self: Tensor, upper: bool=True, *, out: Optional[Tensor]=None) -> Tensor: ...
  914. def potri(self: Tensor, upper: bool=True, *, out: Optional[Tensor]=None) -> Tensor: ...
  915. def potrs(self: Tensor, input2: Tensor, upper: bool=True, *, out: Optional[Tensor]=None) -> Tensor: ...
  916. @overload
  917. def pow(self: Tensor, exponent: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  918. @overload
  919. def pow(self: Union[builtins.float, builtins.int], exponent: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  920. @overload
  921. def pow(self: Tensor, exponent: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  922. @overload
  923. def prod(self: Tensor, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  924. @overload
  925. def prod(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  926. @overload
  927. def prod(self: Tensor, dim: builtins.int, keepdim: bool, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  928. @overload
  929. def prod(self: Tensor, dim: builtins.int, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tensor: ...
  930. @overload
  931. def prod(self: Tensor, dim: builtins.int, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  932. def pstrf(self: Tensor, upper: bool=True, tol: Union[builtins.float, builtins.int]=-1, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  933. def qr(self: Tensor, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  934. @overload
  935. def rand(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  936. @overload
  937. def rand(*size: builtins.int, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  938. @overload
  939. def rand(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, generator: Generator, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  940. @overload
  941. def rand(*size: builtins.int, generator: Generator, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  942. @overload
  943. def rand_like(self: Tensor) -> Tensor: ...
  944. @overload
  945. def rand_like(self: Tensor, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  946. @overload
  947. def randint(high: builtins.int, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  948. @overload
  949. def randint(high: builtins.int, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, generator: Generator, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  950. @overload
  951. def randint(low: builtins.int, high: builtins.int, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  952. @overload
  953. def randint(low: builtins.int, high: builtins.int, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, generator: Generator, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  954. @overload
  955. def randint_like(self: Tensor, high: builtins.int) -> Tensor: ...
  956. @overload
  957. def randint_like(self: Tensor, low: builtins.int, high: builtins.int) -> Tensor: ...
  958. @overload
  959. def randint_like(self: Tensor, high: builtins.int, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  960. @overload
  961. def randint_like(self: Tensor, low: builtins.int, high: builtins.int, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  962. @overload
  963. def randn(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  964. @overload
  965. def randn(*size: builtins.int, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  966. @overload
  967. def randn(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, generator: Generator, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  968. @overload
  969. def randn(*size: builtins.int, generator: Generator, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  970. @overload
  971. def randn_like(self: Tensor) -> Tensor: ...
  972. @overload
  973. def randn_like(self: Tensor, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  974. @overload
  975. def randperm(n: builtins.int, *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  976. @overload
  977. def randperm(n: builtins.int, *, generator: Generator, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  978. @overload
  979. def range(start: Union[builtins.float, builtins.int], end: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  980. @overload
  981. def range(start: Union[builtins.float, builtins.int], end: Union[builtins.float, builtins.int], step: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  982. def reciprocal(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  983. def relu_(self: Tensor) -> Tensor: ...
  984. @overload
  985. def remainder(self: Tensor, other: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  986. @overload
  987. def remainder(self: Tensor, other: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  988. def renorm(self: Tensor, p: Union[builtins.float, builtins.int], dim: builtins.int, maxnorm: Union[builtins.float, builtins.int], *, out: Optional[Tensor]=None) -> Tensor: ...
  989. def reshape(self: Tensor, shape: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> Tensor: ...
  990. def rfft(self: Tensor, signal_ndim: builtins.int, normalized: bool=False, onesided: bool=True) -> Tensor: ...
  991. def rot90(self: Tensor, k: builtins.int=1, dims: Union[Tuple[builtins.int, ...], List[builtins.int], Size]=(0,1)) -> Tensor: ...
  992. def round(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  993. def rrelu_(self: Tensor, lower: Union[builtins.float, builtins.int]=0.125, upper: Union[builtins.float, builtins.int]=0.3333333333333333, training: bool=False, generator: Generator=None) -> Tensor: ...
  994. def rsqrt(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  995. def save(obj, f, pickle_module=pickle, pickle_protocol=2): ...
  996. def selu_(self: Tensor) -> Tensor: ...
  997. def set_default_dtype(d): ...
  998. def set_default_tensor_type(t): ...
  999. def set_flush_denormal(mode: bool) -> bool: ...
  1000. def set_num_threads(num: builtins.int) -> None: ...
  1001. def set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None): ...
  1002. def set_rng_state(new_state): ...
  1003. def sigmoid(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1004. def sign(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1005. def sin(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1006. def sinh(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1007. def slogdet(self: Tensor) -> Tuple[Tensor, Tensor]: ...
  1008. def sort(self: Tensor, dim: builtins.int=-1, descending: bool=False, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  1009. @overload
  1010. def sparse_coo_tensor(indices: Tensor, values: Tensor) -> Tensor: ...
  1011. @overload
  1012. def sparse_coo_tensor(indices: Tensor, values: Tensor, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> Tensor: ...
  1013. @overload
  1014. def sparse_coo_tensor(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  1015. @overload
  1016. def sparse_coo_tensor(*size: builtins.int, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  1017. @overload
  1018. def sparse_coo_tensor(indices: Tensor, values: Tensor, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  1019. @overload
  1020. def sparse_coo_tensor(indices: Tensor, values: Tensor, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  1021. def split(tensor:Tensor, split_size_or_sections:Union[List[builtins.int], builtins.int], dim:builtins.int=0) -> List[Tensor]: ...
  1022. def sqrt(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1023. @overload
  1024. def squeeze(self: Tensor) -> Tensor: ...
  1025. @overload
  1026. def squeeze(self: Tensor, dim: builtins.int) -> Tensor: ...
  1027. @overload
  1028. def sspaddmm(beta: Union[builtins.float, builtins.int], self: Tensor, alpha: Union[builtins.float, builtins.int], mat1: Tensor, mat2: Tensor) -> Tensor: ...
  1029. @overload
  1030. def sspaddmm(beta: Union[builtins.float, builtins.int], self: Tensor, mat1: Tensor, mat2: Tensor) -> Tensor: ...
  1031. def stack(tensors: Union[Tuple[Tensor, ...],List[Tensor]], dim: builtins.int=0, *, out: Optional[Tensor]=None) -> Tensor: ...
  1032. @overload
  1033. def std(self: Tensor, unbiased: bool=True, *, out: Optional[Tensor]=None) -> Tensor: ...
  1034. @overload
  1035. def std(self: Tensor, dim: builtins.int, unbiased: bool=True, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tensor: ...
  1036. def stft(input, n_fft, hop_length=None, win_length=None, window=None, center=True, pad_mode='reflect', normalized=False, onesided=True): ...
  1037. @overload
  1038. def sub(self: Tensor, alpha: Union[builtins.float, builtins.int], other: Tensor) -> Tensor: ...
  1039. @overload
  1040. def sub(self: Tensor, alpha: Union[builtins.float, builtins.int], other: Tensor, *, out: Tensor) -> Tensor: ...
  1041. @overload
  1042. def sum(self: Tensor, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  1043. @overload
  1044. def sum(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1045. @overload
  1046. def sum(self: Tensor, dim: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size], keepdim: bool, *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  1047. @overload
  1048. def sum(self: Tensor, dim: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size], keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tensor: ...
  1049. @overload
  1050. def sum(self: Tensor, dim: Union[builtins.int, Tuple[builtins.int, ...], List[builtins.int], Size], *, dtype: _dtype, out: Optional[Tensor]=None) -> Tensor: ...
  1051. def svd(self: Tensor, some: bool=True, compute_uv: bool=True, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor, Tensor]: ...
  1052. def symeig(self: Tensor, eigenvectors: bool=False, upper: bool=True, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  1053. def t(self: Tensor) -> Tensor: ...
  1054. def take(self: Tensor, index: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1055. def tan(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1056. def tanh(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1057. @overload
  1058. def tensor(data: Any, dtype: Optional[_dtype]=None, device: Union[device, str, None]=None, requires_grad: bool=False) -> Tensor: ...
  1059. @overload
  1060. def tensor(storage: Storage, storageOffset: builtins.int, size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], stride: Union[Tuple[builtins.int, ...], List[builtins.int], Size]=()) -> Tensor: ...
  1061. @overload
  1062. def tensor(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], stride: Union[Tuple[builtins.int, ...], List[builtins.int], Size]) -> Tensor: ...
  1063. def tensordot(a:Tensor, b:Tensor, dims=2) -> Tensor: ...
  1064. def topk(self: Tensor, k: builtins.int, dim: builtins.int=-1, largest: bool=True, sorted: bool=True, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  1065. def trace(self: Tensor) -> Union[builtins.float, builtins.int]: ...
  1066. def transpose(self: Tensor, dim0: builtins.int, dim1: builtins.int) -> Tensor: ...
  1067. def tril(self: Tensor, diagonal: builtins.int=0, *, out: Optional[Tensor]=None) -> Tensor: ...
  1068. def triu(self: Tensor, diagonal: builtins.int=0, *, out: Optional[Tensor]=None) -> Tensor: ...
  1069. def trtrs(self: Tensor, A: Tensor, upper: bool=True, transpose: bool=False, unitriangular: bool=False, *, out: Optional[Tensor]=None) -> Tuple[Tensor, Tensor]: ...
  1070. def trunc(self: Tensor, *, out: Optional[Tensor]=None) -> Tensor: ...
  1071. def unbind(self: Tensor, dim: builtins.int=0) -> Union[Tuple[Tensor, ...],List[Tensor]]: ...
  1072. def unique(input, sorted=False, return_inverse=False, dim=None): ...
  1073. def unsqueeze(self: Tensor, dim: builtins.int) -> Tensor: ...
  1074. @overload
  1075. def var(self: Tensor, unbiased: bool=True, *, out: Optional[Tensor]=None) -> Tensor: ...
  1076. @overload
  1077. def var(self: Tensor, dim: builtins.int, unbiased: bool=True, keepdim: bool=False, *, out: Optional[Tensor]=None) -> Tensor: ...
  1078. def where(condition: Tensor, self: Tensor, other: Tensor) -> Tensor: ...
  1079. @overload
  1080. def zeros(size: Union[Tuple[builtins.int, ...], List[builtins.int], Size], *, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  1081. @overload
  1082. def zeros(*size: builtins.int, out: Optional[Tensor]=None, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  1083. @overload
  1084. def zeros_like(self: Tensor) -> Tensor: ...
  1085. @overload
  1086. def zeros_like(self: Tensor, *, dtype: _dtype=None, layout: layout=strided, device: Optional[device]=None, requires_grad:bool=False) -> Tensor: ...
  1087. class DoubleStorage(Storage): ...
  1088. class FloatStorage(Storage): ...
  1089. class LongStorage(Storage): ...
  1090. class IntStorage(Storage): ...
  1091. class ShortStorage(Storage): ...
  1092. class CharStorage(Storage): ...
  1093. class ByteStorage(Storage): ...
  1094. class DoubleTensor(Tensor): ...
  1095. class FloatTensor(Tensor): ...
  1096. class LongTensor(Tensor): ...
  1097. class IntTensor(Tensor): ...
  1098. class ShortTensor(Tensor): ...
  1099. class CharTensor(Tensor): ...
  1100. class ByteTensor(Tensor): ...
  1101. complex128: dtype = ...
  1102. complex32: dtype = ...
  1103. complex64: dtype = ...
  1104. double: dtype = ...
  1105. float: dtype = ...
  1106. float16: dtype = ...
  1107. float32: dtype = ...
  1108. float64: dtype = ...
  1109. half: dtype = ...
  1110. int: dtype = ...
  1111. int16: dtype = ...
  1112. int32: dtype = ...
  1113. int64: dtype = ...
  1114. int8: dtype = ...
  1115. long: dtype = ...
  1116. short: dtype = ...
  1117. uint8: dtype = ...