Skip to main content

TENSOR_STATE_HIDDEN_STATES

Constant TENSOR_STATE_HIDDEN_STATES 

Source
pub const TENSOR_STATE_HIDDEN_STATES: &str = "state_hidden_states";
Expand description

State エンコーダが出力し、Query 評価器が参照する最終層隠れ状態テンソル名 (float32[batch_size, state_seq_len, hidden_size])。