(batch_size: int, penalty_value: float)
| 40 | |
| 41 | |
| 42 | def _create_penalty_tensor(batch_size: int, penalty_value: float) -> paddle.Tensor: |
| 43 | return paddle.full(shape=[batch_size, 1], fill_value=penalty_value, dtype="float32") |
| 44 | |
| 45 | |
| 46 | def _create_tokens_tensor( |
no outgoing calls
no test coverage detected