tensorflow-federated - 有没有办法转换联合值？

Question

如果我有联合值，请说 {int32}@CLIENTS我想转换到 {float32}@CLIENTS是否有捷径可寻？谢谢!

Answer 1

张量操作通常需要发生在装饰有 tff.tf_computation 的函数内部。 .由于提到的类型有展示位置( @CLIENTS )，这可能在 tff.federated_computation 内装饰函数，因此需要使用 tff.federated_map 调用转换方法.
像这样的东西:

@tff.tf_computation
def cast_to_float(x):
  return tf.cast(x, tf.float32)

@tff.federated_computation(tff.FederatedType(int32, tff.CLIENTS))
def my_func(a):
  a_float = tff.federated_map(cast_to_float, a)
  return a_float

print(my_func.type_signature)

>>> ({int32}@CLIENTS -> {float32}@CLIENTS)