get_bitmask_shape

Function get_bitmask_shape 

Source
pub fn get_bitmask_shape(batch_size: usize, vocab_size: usize) -> (usize, usize)
Expand description

Return the shape of the bitmask: (batch_size, ceil(vocab_size/32)).

Port of the vendored matcher::get_bitmask_shape.