def test_get_latest_tensor_shapes():
lstm = LSTMClassifier(embedding_dim=embedding_dim,
hidden_dim=hidden_dim,
vocab_size=vocab_size)
input_tensor = torch.randint(low=0,
high=5,
size=(batch_size, 1 + sequence_length),
dtype=torch.long)
output = lstm(input_tensor)
tensor_shapes = lstm.get_latest_tensor_shapes()
global tensors
tensors = tensor_shapes
assert (len(tensor_shapes) == 12)
print("Test passed for *test_get_latest_tensor_shapes*\n")
