Skip to main content
TensorFlow
Install Learn
  • Introduction
    New to TensorFlow?
  • Tutorials
    Learn how to use TensorFlow with end-to-end examples
  • Guide
    Learn framework concepts and components
  • Learn ML
    Educational resources to master your path with TensorFlow
API
  • TensorFlow (v2.16.1)
  • Versions…
  • TensorFlow.js
  • TensorFlow Lite
  • TFX
Ecosystem
  • LIBRARIES
  • TensorFlow.js
    Develop web ML applications in JavaScript
  • TensorFlow Lite
    Deploy ML on mobile, microcontrollers and other edge devices
  • TFX
    Build production ML pipelines
  • All libraries
    Create advanced models and extend TensorFlow
  • RESOURCES
  • Models & datasets
    Pre-trained models and datasets built by Google and the community
  • Tools
    Tools to support and accelerate TensorFlow workflows
  • Responsible AI
    Resources for every stage of the ML workflow
  • Recommendation systems
    Build recommendation systems with open source tools
Community
  • Groups
    User groups, interest groups and mailing lists
  • Contribute
    Guide for contributing to code and documentation
  • Blog
    Stay up to date with all things TensorFlow
  • Forum
    Discussion platform for the TensorFlow community
Why TensorFlow
  • About
  • Case studies
/
  • English
  • 中文 – 简体
GitHub Sign in
  • TensorFlow v2.16.1
Overview Python C++ Java More
TensorFlow
  • Install
  • Learn
    • More
  • API
    • More
    • Overview
    • Python
    • C++
    • Java
    • More
  • Ecosystem
    • More
  • Community
    • More
  • Why TensorFlow
    • More
  • GitHub
  • Overview
  • TensorFlow Models
  • tfm
    • Overview
  • tfm.core
    • Overview
    • actions
      • Overview
      • EMACheckpointing
      • PruningAction
      • RecoveryAction
      • RecoveryCondition
      • get_eval_actions
      • get_train_actions
    • base_task
      • Overview
      • DifferentialPrivacyConfig
      • RuntimeConfig
      • Task
    • base_trainer
      • Overview
      • ExperimentConfig
      • Trainer
      • TrainerConfig
      • get_runtime_options
    • config_definitions
      • Overview
      • DataConfig
      • TaskConfig
    • exp_factory
      • Overview
      • get_exp_config
      • register_config_factory
    • export_base
      • Overview
      • ExportModule
      • export
      • get_timestamped_export_dir
    • file_writers
      • Overview
      • write_small_dataset
    • input_reader
      • Overview
      • InputReader
      • match_files
    • registry
      • Overview
      • lookup
      • register
    • savedmodel_checkpoint_manager
      • Overview
      • SavedModelCheckpointManager
      • make_saved_modules_directory_name
    • task_factory
      • Overview
      • get_task
      • get_task_cls
      • register_task_cls
    • tf_example_builder
      • Overview
      • BytesValueType
      • TfExampleBuilder
    • tf_example_feature_key
      • Overview
      • TfExampleFeatureKeyBase
      • dataclass
    • train_lib
      • Overview
      • OrbitExperimentRunner
      • maybe_create_best_ckpt_exporter
      • run_experiment
    • train_utils
      • Overview
      • BestCheckpointExporter
      • ExperimentParser
      • ParseConfigOptions
      • cast_leaf_nested_dict
      • convert_variables_to_constants_v2_as_graph
      • create_optimizer
      • create_trainer
      • get_leaf_nested_dict
      • parse_configuration
      • read_global_step_from_checkpoint
      • remove_ckpts
      • save_gin_config
      • serialize_config
      • try_count_flops
      • try_count_params