當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.distribute.experimental.rpc.Server.create用法及代碼示例


用法

@staticmethod
create(
    rpc_layer, address
)

參數

  • rpc_layer 客戶端和服務器之間的通信層。目前僅支持"grpc" rpc 層。
  • address 托管 RPC 服務器的地址。

返回

拋出

  • 如果使用 rpc_layer 而不是 "grpc",則會出現 ValueError。目前僅支持 GRPC。

在給定地址創建 TF RPC 服務器。

示例用法:

import portpicker
@tf.function(input_signature=[
     tf.TensorSpec([], tf.int32),
     tf.TensorSpec([], tf.int32)])
def remote_fn(a, b):
  return tf.add(a, b)
port = portpicker.pick_unused_port()
address = "localhost:{}".format(port)
server = tf.distribute.experimental.rpc.Server.create("grpc", address)
server.register("addition", remote_fn)
server.start()

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.distribute.experimental.rpc.Server.create。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。