jax.profiler.start_server# jax.profiler.start_server(port)[源代码][源代码]# 在端口 port 启动分析器服务器。 在 TensorBoard 2.2 或更高版本中使用“TensorFlow 分析器”功能,您可以连接到分析器服务器并采样显示 CPU、GPU 和/或 TPU 设备活动的执行轨迹。 参数: port (int) 返回类型: xla_client.profiler.ProfilerServer