Function - split_tensor_dict_into_floats_and_non_floats

Function - split_tensor_dict_into_floats_and_non_floats#

openfl.utilities.split.split_tensor_dict_into_floats_and_non_floats(tensor_dict)[source]#

Split the tensor dictionary into float and non-floating point values.

This function splits a tensor dictionary into two dictionaries: one containing all the floating point tensors and the other containing all the non-floating point tensors.

Parameters:

tensor_dict (dict) – A dictionary of tensors.

Returns:

The first dictionary contains all of the floating

point tensors and the second dictionary contains all of the non-floating point tensors.

Return type:

Tuple[dict, dict]