like use `torch._foreach_mul_` `torch._foreach_add_` ` torch._foreach_addcmul_` `torch._foreach_sqrt` `torch._foreach_zeros_like`