machine-learning - torch.nn.ReflectionPad2d 仅适用于浮动

Question

它只能在浮动下工作吗？当我尝试了不同的 dtype，如 int32、Long 和 Byte 时，它​​似乎只适用于 dtype=torch.float。
例如：

m = nn.ReflectionPad2d(2)
tensor = torch.arange(9, dtype=torch.int8).reshape(1, 1, 3, 3)
m(tensor)

RuntimeError：“char”未实现“reflection_pad2d”

标签： machine-learningdeep-learningpytorchpaddinggenerative-adversarial-network