torch/nn/modules/flatten.pyi.in

from typing import Any
from .module import Module

class Flatten(Module):
    __constants__: Any = ...
    start_dim: Any = ...
    end_dim: Any = ...
    def __init__(self, start_dim: int = ..., end_dim: int = ...) -> None: ...
    def forward(self, input: Any): ...