You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
9 lines
296 B
Python
9 lines
296 B
Python
from torch.nn import PixelUnshuffle as _PixelUnshuffle
|
|
|
|
from imaginairy.vendored.refiners.fluxion.layers.module import Module
|
|
|
|
|
|
class PixelUnshuffle(_PixelUnshuffle, Module):
|
|
def __init__(self, downscale_factor: int):
|
|
_PixelUnshuffle.__init__(self, downscale_factor=downscale_factor)
|