python - AttributeError：模块“torch.nn”没有属性“ReflectionPad3d”

Question

我正在练习 PyTorch 中的填充层。

1d和2d反射填充效果很好。

当我尝试运行3d padding 中给出的示例时，发生了标题中给出的错误。

m = nn.ReflectionPad3d(1)
input = torch.arange(8, dtype=torch.float).reshape(1, 1, 2, 2, 2)
m(input)

这个错误的原因是什么？

标签： pythonpytorch

Answer 1

不幸的是ReflectionPad3d，官方版本中还没有。您所指的文档针对的是不稳定的开发者预览版。查看最新稳定版本 1.9.0 的填充层部分，看看哪些是可用层。由于该主题的官方问题已经结束，我相信它将进入下一个版本。