docs: def transpose(self, name=None):
      """Returns transpose batch reshape."""
      if name is None:
        name = self.module_name + "_transpose"
      return BatchReshape(shape=lambda: self.input_shape,
                          preserve_dims=self._preserve_dims,
                          name=name)