class_docstr: A 2d-Unet implementation with sane defaults.
class_docstr_tok: ["A","2d-Unet","implementation","with","sane","defaults","."]
code: def forward(self, x):
      in_layer = self.in_layer(x)
      unet = self.unet_block(in_layer)
      out_layer = self.out_layer(unet)
      return out_layer
  
imports: import torch
  import torch.nn
  import torchvision.utils
  from torch.nn import functional
label_desc: Use image node_id 2 for calling the Unet obj's underlying member method code with example usage: obj.forward(x) and returns: out_layer