跳到主要内容
Ctrl+K
本站点涵盖了新的 Flax NNX API。[点击此处查看旧的Flax Linen API]
 - Home
  • Flax 基础
  • MNIST 教程
  • 为什么选择 Flax NNX?
  • 指南
    • 转换
    • 在多个设备上扩展
    • 使用过滤器
    • 随机性
    • 性能考虑
    • 从 Flax Linen 到 NNX 的演变
    • 一起使用 Flax NNX 和 Linen
    • 模型手术
    • 保存和加载检查点
    • Flax NNX 与 JAX 转换
    • 从 Haiku 迁移到 Flax
    • 示例:使用预训练的 Gemma 通过 Flax NNX 进行推理
  • 示例
    • 核心示例
  • Flax NNX 词汇表
  • Flax 的理念
  • 如何贡献
  • API 参考
    • flax.nnx
      • graph
      • module
      • nn
        • 激活函数
        • 注意力机制
        • 数据类型
        • 初始化器
        • 线性层
        • LoRA
        • 归一化
        • 循环层
        • 随机
      • rnglib
      • spmd
      • 状态
      • 训练
        • 指标
        • 优化器
      • 转换
      • 变量
      • 助手
      • 可视化
      • filterlib
      • 桥接
    • flax.core.frozen_dict 包
    • flax.struct 包
    • flax.training 包
  • .rst

filterlib

目录

  • to_predicate()
  • WithTag
  • PathContains
  • OfType
  • Any
  • All
  • Not
  • Everything
  • Nothing

filterlib#

flax.nnx.filterlib.to_predicate(filter)[源代码]#

将过滤器转换为谓词函数。请参阅 使用过滤器。

class flax.nnx.WithTag(tag: str)[源代码]#
class flax.nnx.PathContains(key: flax.typing.Key)[源代码]#
class flax.nnx.OfType(type: type)[源代码]#
class flax.nnx.Any(*filters)[源代码]#
class flax.nnx.All(*filters)[源代码]#
class flax.nnx.Not(collection_filter, /)[源代码]#
class flax.nnx.Everything[源代码]#
class flax.nnx.Nothing[源代码]#

上一页

可视化

下一页

桥接

目录
  • to_predicate()
  • WithTag
  • PathContains
  • OfType
  • Any
  • All
  • Not
  • Everything
  • Nothing

Flax 作者

© 版权所有 2023,Flax 作者。