Python torch.nn 模块,MaxUnpool3d() 实例源码

我们从Python开源项目中,提取了以下1个代码示例,用于说明如何使用torch.nn.MaxUnpool3d()

项目:pytorch-dist    作者:apaszke    | 项目源码 | 文件源码
def __init__(self):
        super(UnpoolingNet3d, self).__init__(
            pool=nn.MaxPool3d(2, return_indices=True),
            unpool=nn.MaxUnpool3d(2)
        )