如果我有一个联合值,比如说我想转换为 {float32}@CLIENTS
的 {int32}@CLIENTS
,有没有一种简单的方法可以做到这?谢谢!
最佳答案
张量操作通常需要发生在用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)
关于tensorflow-federated - 有没有办法类型转换联合值(value)?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62598087/