class_docstr: Root mean square pooling
code: def forward(self, x):  # skipcq: PYL-W0221
      x_mean = x.mean(dim=[2, 3])
      avg_pool = self.avg_pool((x - x_mean) ** 2)
      return avg_pool.sqrt()
  
imports: from typing import Union, Dict
  import torch
  import torch.nn.functional
  from torch.nn.modules.module import _IncompatibleKeys
  from torch import Tensor, nn
label_desc: Use image node_id 2 for calling the RMSPool obj's underlying member method code with example usage: obj.forward(x) and returns: avg_pool