tff.federated_map

Maps a federated value pointwise using a mapping function.

tff.federated_map(
    fn, arg
)

The function fn is applied separately across the group of devices represented by the placement type of arg. For example, if value has placement type tff.CLIENTS, then fn is applied to each client individually. In particular, this operation does not alter the placement of the federated value.

fn A mapping function to apply pointwise to member constituents of arg. The parameter of this function must be of the same type as the member constituents of arg.
arg A value of a TFF federated type (or a value that can be implicitly converted into a TFF federated type, e.g., by zipping) placed at tff.CLIENTS or tff.SERVER.

A federated value with the same placement as arg that represents the result of fn on the member constituent of arg.

TypeError If the arguments are not of the appropriate types.