mpnet_rs::mpnet

Function create_position_ids_from_input_ids

Source 
pub fn create_position_ids_from_input_ids(
    input_ids: &Tensor,
    padding_idx: u32,
) -> Result<Tensor>
Expand description

Creates position ids from the input ids.

§Arguments

  • input_ids - The input ids tensor.
  • padding_idx - The padding index.

§Returns

  • Tensor - The position ids tensor.