class_docstr: Interface for `EmbeddingCollection`.
code: def forward(
      self,
      features: KeyedJaggedTensor,
  ) -> Dict[str, JaggedTensor]:
      pass
  
imports: import abc
  from typing import Dict, List, Optional, Tuple, Union
  import torch
  import torch.nn
  from torchrec.modules.embedding_configs import DataType, EmbeddingBagConfig, EmbeddingConfig, pooling_type_to_str
  from torchrec.sparse.jagged_tensor import JaggedTensor, KeyedJaggedTensor, KeyedTensor
label_desc: Use image node_id 1 for calling the EmbeddingCollectionInterface obj's underlying member method code with example usage: obj.forward(features) without return types