class_docstr: Module to embed Bag data type, works on input of token frequencies.
class_docstr_tok: ["Module","to","embed","Bag","data","type",",","works","on","input","of","token","frequencies","."]
code: def output_shape(self) -> torch.Size:
      return torch.Size([self.embedding_size])
  
imports: import logging
  from typing import Dict, List, Optional, Tuple, Union
  import torch
  from torch import nn
  from ludwig.constants import TYPE
  from ludwig.modules.initializer_modules import get_initializer
  from ludwig.utils.data_utils import load_pretrained_embeddings
  from ludwig.utils.torch_utils import get_torch_device, LudwigModule
label_desc: Use image node_id 4 for calling the EmbedWeighted obj's underlying member method code with example usage: obj.output_shape() and returns: torch