Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
fold.pyi 1.1 KB
Newer Older
Mandeep Singh Baines's avatar
Initial commit  
Mandeep Singh Baines committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34 
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.

from .module import Module
from ... import Tensor
from ..common_types import _size_any_t


class Fold(Module):
    output_size: _size_any_t = ...
    kernel_size: _size_any_t = ...
    dilation: _size_any_t = ...
    padding: _size_any_t = ...
    stride: _size_any_t = ...

    def __init__(self, output_size: _size_any_t, kernel_size: _size_any_t, dilation: _size_any_t = ...,
                 padding: _size_any_t = ..., stride: _size_any_t = ...) -> None: ...

    def forward(self, input: Tensor) -> Tensor: ...  # type: ignore

    def __call__(self, input: Tensor) -> Tensor: ...  # type: ignore


class Unfold(Module):
    kernel_size: _size_any_t = ...
    dilation: _size_any_t = ...
    padding: _size_any_t = ...
    stride: _size_any_t = ...

    def __init__(self, kernel_size: _size_any_t, dilation: _size_any_t = ..., padding: _size_any_t = ...,
                 stride: _size_any_t = ...) -> None: ...

    def forward(self, input: Tensor) -> Tensor: ...  # type: ignore

    def __call__(self, input: Tensor) -> Tensor: ...  # type: ignore