code: def forward(self, images):
      x = self.block(images)
      if self.cnn_num == 2:
          # (b, w, h, c)
          x = paddle.transpose(x, [0, 3, 2, 1])
          x_shape = paddle.shape(x)
          x = paddle.reshape(
              x, [x_shape[0], x_shape[1], x_shape[2] * x_shape[3]]
          )
      return x
  
imports: from paddle import nn
  import paddle
label_desc: Use image node_id 2 for calling the MTB obj's underlying member method code with example usage: obj.forward(images) and returns: x