自定义# 创建运算 前提条件 定义运算接口 实现运算的内核 多线程 CPU 内核 GPU 内核 构建运算库 使用系统编译器编译运算(TensorFlow 二进制文件安装) 使用 Bazel 编译运算(TensorFlow 源代码安装) 在 Python 中使用运算 验证运算是否正常运行 将高级功能构建到运算中 条件检查和验证 运算注册 特性 特性类型 默认值和约束 多态性 类型多态性 命名 类型多态化示例 列表输入和输出 输入和输出 向后兼容性 GPU 支持 编译 GPU 设备的内核 在 Python 中实现梯度 C++ 中的形状函数 为您的自定义运算构建 pip 软件包 生成随机数 设置 tf.random.Generator 类 创建独立的随机数流 与 tf.function 交互 在 tf.function 的外部创建生成器 在 tf.function 的内部创建生成器 将生成器作为参数传递给 tf.function 与分布策略交互 在分布策略的外部创建生成器 在分布策略的内部创建生成器 保存生成器 检查点 SavedModel 无状态 RNG 算法 基本信息 XLA 设备