Sign in
android
/
platform
/
external
/
pytorch
/
e5cf7afd0a
/
.
/
torch
/
nn
/
modules
/
pixelshuffle.pyi.in
blob: bd1b4055e0739591cdcc3beae25c748c04c00ae7 [
file
] [
log
] [
blame
]
from
.
module
import
Module
from
...
import
Tensor
class
PixelShuffle
(
Module
):
upscale_factor
:
int
=
...
def
__init__
(
self
,
upscale_factor
:
int
)
->
None
:
...
def
forward
(
self
,
input
:
Tensor
)
->
Tensor
:
...
