def test_from_tensor(self):
self.assertEqual(
EntityList.from_tensor(torch.tensor([3, 4], dtype=torch.long)),
EntityList(torch.tensor([3, 4], dtype=torch.long),
TensorList.empty(num_tensors=2)),
)
def from_tensor(cls, tensor: LongTensorType) -> 'EntityList':
if tensor.ndimension() != 1:
raise ValueError("Expected 1D tensor, got %dD" %
tensor.ndimension())
tensor_list = TensorList.empty(num_tensors=tensor.nelement())
return cls(tensor, tensor_list)
def test_empty(self):
self.assertEqual(
EntityList.empty(),
EntityList(torch.empty((0, ), dtype=torch.long),
TensorList.empty()),
)
def empty(cls) -> 'EntityList':
return cls(torch.empty((0, ), dtype=torch.long), TensorList.empty())