Esempi in Python per TensorList.cat

Esempio n. 1

File: entitylist.py Progetto: RweBs/PDKE

 def cat(cls, entity_lists: Sequence['EntityList']) -> 'EntityList':
     return cls(torch.cat([el.tensor for el in entity_lists]),
                TensorList.cat(el.tensor_list for el in entity_lists))

Esempio n. 2

File: entitylist.py Progetto: adamlerer/PyTorch-BigGraph-1

 def cat(cls, entity_lists: Sequence["EntityList"]) -> "EntityList":
     return cls(
         torch.cat([el.tensor for el in entity_lists]),
         TensorList.cat(el.tensor_list for el in entity_lists),
     )