變數詞彙表

變數字典是一個一般的 Python 字典，作為一個或多個「變數集合」的容器，它們各自都是巢狀字典，其葉子是 jax.numpy 陣列。

不同的變數集合分享相同的巢狀樹狀結構。

例如，考量以下變數字典

{
  "params": {
    "Conv1": { "weight": ..., "bias": ... },
    "BatchNorm1": { "scale": ..., "mean": ... },
    "Conv2": {...}
  },
  "batch_stats": {
    "BatchNorm1": { "moving_mean": ..., "moving_average": ...}
  }
}

在此情況下，"BatchNorm1" 鍵位於 "params" 與 `"batch_stats"" 集中。這反映了名為 ""BatchNorm1"" 的子模組同時具備可訓練參數（"params" 集）及其他不可訓練變數（"batch_stats" 集）的事實。

待辦事項：製作「變數字典」設計註記，並在此處連結至該註記。

class flax.linen.Variable(scope, collection, name, unbox)

「變數」物件允許可變存取「變數字典」中的變數。

變數以集合（例如「batch_stats」）與名稱（例如「moving_mean」）識別。value 屬性可存取變數的內容，並可指定給予變數以進行變更。