code: def __init__(
      self, summarization_type: Literal["sum", "mean", "max", "min"]
  ):
      super().__init__()
      assert summarization_type in {"sum", "mean", "max", "min"}
      self.__summarization_type = summarization_type
  
imports: from typing_extensions import Literal
  import torch
  from abc import abstractmethod
  from math import sqrt
  from torch import nn
  from torch_scatter import scatter, scatter_log_softmax, scatter_sum
  from typing import NamedTuple, Union
label_desc: Use image node_id 1 to create a new SimpleVarSizedElementReduce object from inherited base classes: AbstractVarSizedElementReduce with example: obj = SimpleVarSizedElementReduce(summarization_type)