jax.nn.get_scaled_dot_general_config

jax.nn.get_scaled_dot_general_config(mode, global_scale=None)

获取 scaled_dot_general 的量化配置。

为 jax.nn.scaled_dot_general 创建量化配置。

另请参阅

• jax.nn.scaled_dot_general()：缩放点积通用函数。

参数:

• mode (Literal['nvfp4', 'mxfp8'])

• global_scale (Array | None | None)