class ModulatedDeformConv(nn.Module):
应用基于NPU的Modulated Deformable 2D卷积操作。
from torch_npu.contrib.module import ModulatedDeformConv m = ModulatedDeformConv(32, 32, 1)
>>> m = ModulatedDeformConv(32, 32, 1) >>> input_tensor = torch.randn(2, 32, 5, 5) >>> output = m(input_tensor) >>> x = torch.randn(2, 32, 7, 7) >>> model = ModulatedDeformConv(32, 32, 3, 2, 1) >>> torch.npu.set_device(0) >>> x = x.npu() >>> model = model.npu() >>> o = model(x) >>> l = o.sum() >>> l.backward() >>> print(l)