資料類型 (Dtypes)#
- flax.nnx.nn.dtypes.canonicalize_dtype(*args, dtype=None, inexact=True)[原始碼]#
將可選的資料類型標準化為最終的資料類型。
如果
dtype為 None，此函數將推斷資料類型。如果它不是 None，它將被原封不動地返回，如果資料類型無效，則會引發異常。從輸入參數使用jnp.result_type推斷資料類型。- 參數
*args – 與 JAX 陣列相容的值。None 值將被忽略。
dtype – 可選的資料類型覆蓋。如果指定，則會將參數轉換為指定的資料類型，並且禁用資料類型推斷。
inexact – 當為 True 時，輸出資料類型必須是一個子資料類型
This (of jnp.inexact。非精確的資料類型為實數或複數浮點數。) –
on (當您想要應用不直接運作的操作時很有用) –
example. (像是求整數的平均值) –
- 回傳值
應該將 *args 轉換成的資料類型。
- flax.nnx.nn.dtypes.promote_dtype(args, /, *, dtype=None, inexact=True)[原始碼]#
“將輸入參數提升為指定的或推斷的資料類型。
所有參數都被轉換為相同的資料類型。請參閱
canonicalize_dtype了解如何確定此資料類型。promote_dtype 的行為主要是在
jax.numpy.promote_types周圍的便利包裝。不同之處在於它會自動將所有輸入轉換為推斷的資料類型，允許通過強制資料類型覆蓋推斷，並有一個可選的檢查來保證結果資料類型是非精確的。- 參數
*args – 與 JAX 陣列相容的值。None 值將按原樣返回。
dtype – 可選的資料類型覆蓋。如果指定，則會將參數轉換為指定的資料類型，並且禁用資料類型推斷。
inexact – 當為 True 時，輸出資料類型必須是一個子資料類型
This (of jnp.inexact。非精確的資料類型為實數或複數浮點數。) –
on (當您想要應用不直接運作的操作時很有用) –
example. (像是求整數的平均值) –
- 回傳值
轉換為相同資料類型的陣列的參數。