class_docstr: Args:
class_docstr_tok: ["Args",":"]
code: def __init__(self, block, n_segment):
      super().__init__()
      self.block = block
      self.nl = NONLocalBlock3D(block.bn3.num_features)
      self.n_segment = n_segment
  
imports: import torch
  from torch import nn
  from torch.nn import functional
  import torchvision
label_desc: Use image node_id 1 to create a new NL3DWrapper object from inherited base classes: nn with example: obj = NL3DWrapper(block, n_segment)