Function _create_fake_logits

tests/layers/test_speculative_sampler.py:37–39

(batch_size: int, vocab_size: int)

def _create_fake_logits(batch_size: int, vocab_size: int) -> paddle.Tensor:
    fake_logits = paddle.rand(shape=[batch_size, vocab_size], dtype="float32")
    return fake_logits


def _create_penalty_tensor(batch_size: int, penalty_value: float) -> paddle.Tensor: