nip.utils.maths.dict_scalar_multiple

nip.utils.maths.dict_scalar_multiple#

nip.utils.maths.dict_scalar_multiple(dictionary: dict[Tensor], scalar: float) dict[Tensor][source]#

Calculate a scalar multiple of a dictionary of tensors.

Parameters:
  • dictionary (dict[Tensor]) – The dictionary of tensors.

  • scalar (float) – The scalar to multiply the dictionary by.

Returns:

scaled_dict (dict[Tensor]) – The scaled dictionary.