blob: 13079b4b6e0bb1f2be0ab7c474500684fe9b9e6f [file] [log] [blame]
from typing import Union, Iterable
from ... import Tensor
_tensor_or_tensors = Union[Tensor, Iterable[Tensor]]
def clip_grad_norm_(parameters: _tensor_or_tensors, max_norm: float, norm_type: float = ...): ...
def clip_grad_value_(parameters: _tensor_or_tensors, clip_value: float): ...