nip.utils.tensordict.tensordict_scalar_multiply#
- nip.utils.tensordict.tensordict_scalar_multiply(td: TensorDictBase, scalar: float | int, *, inplace=False) TensorDictBase [source]#
Multiply a tensordict by a scalar.
- Parameters:
- Returns:
td_scaled (TensorDictBase) – The scaled tensordict