Projekt_AI-Automatyczny_saper/venv/Lib/site-packages/torch/nn/parameter.pyi
2021-06-01 17:38:31 +02:00

16 lines
464 B
Python

import torch
from .. import Tensor
from typing import Tuple, Optional
import builtins
class Parameter(Tensor):
def __init__(self, data: Tensor=..., requires_grad: builtins.bool=...): ...
...
class UninitializedParameter(Tensor):
def __init__(self, data: Tensor=..., requires_grad: builtins.bool=...): ...
def materialize(self, shape: Tuple[int, ...], device: Optional[torch.device] = None, dtype: Optional[torch.dtype] = None): ...
...