tff.federated_map

Maps a federated value pointwise using a mapping function.

tff.federated_map(
    fn, arg
)

Used in the notebooks

Used in the tutorials
Implementing Custom Aggregations Sending Different Data To Particular Clients With tff.federated_select Building Your Own Federated Learning Algorithm Custom Federated Algorithms, Part 1: Introduction to the Federated Core Custom Federated Algorithms, Part 2: Implementing Federated Averaging

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.

Args
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.

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

Raises
TypeError	If the arguments are not of the appropriate types.