当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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

用法

@staticmethod
create(
    rpc_layer, address
)

参数

rpc_layer 客户端和服务器之间的通信层。目前仅支持"grpc" rpc 层。
address 托管 RPC 服务器的地址。

返回

tf.distribute.experimental.rpc.Server 类的实例。

抛出

如果使用 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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。