code: def colorize(self, rowwise=False):
      self._colorize = (
          Colorize.ROWWISE if rowwise else Colorize.COLUMNWISE
      )
  
imports: import collections
  import enum
  import itertools
  from typing import DefaultDict, List, Optional, Tuple
  from torch.utils.benchmark.utils import common
  from torch import tensor
label_desc: Use image node_id 5 for calling the Compare obj's underlying member method code with example usage: obj.colorize(rowwise) without return types