Struct ModelInputDimensions 

pub struct ModelInputDimensions {
    pub batch_size: usize,
    pub sequence_length: usize,
    pub num_options: usize,
}

検証済みのモデル入力次元情報。

Fields

batch_size: usize

バッチサイズ(質問並列数)。

sequence_length: usize

入力系列長(パディング含む)。

num_options: usize

評価候補数。

Implementations

impl ModelInputDimensions

pub fn new( batch_size: usize, sequence_length: usize, num_options: usize, ) -> Result<Self>

新規次元情報を構築し、仕様制約を満たしているか検証する。

pub fn validate_tensor_shapes( input_ids_shape: &[usize], attention_mask_shape: &[usize], op_indices_shape: &[usize], ) -> Result<Self>

各入力テンソルの形状スライスを検証し、一致する次元情報を返却する。

pub fn validate_output_shape(&self, logits_shape: &[usize]) -> Result<()>

出力 logits テンソルの形状が入力次元と整合しているか検証する。

Trait Implementations

impl Clone for ModelInputDimensions

fn clone(&self) -> ModelInputDimensions

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for ModelInputDimensions

impl Debug for ModelInputDimensions

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Eq for ModelInputDimensions

impl PartialEq for ModelInputDimensions

fn eq(&self, other: &ModelInputDimensions) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for ModelInputDimensions