Выражение, которое возвращает измененный список

Question

Я ищу одно выражение, мутирую элемент и возвращаю измененный список

Далее немного подробно

      # key=0; value=3; rng=[1,2]
[(v if i != key else value) for i, v in enumerate(rng)]

---

Изменить :

Я ищу способ встроить следующую функцию в одно выражение

      def replace(rng: List, key: int, value):
    a = list(rng) 
    a[key] = value
    return a

---

Редактировать 2 : код, который фактически мотивировал этот вопрос

      class TextDecoder(nn.Module):
    def forward(self, x: Tensor, kv_cache: Tensor):
        kv_cache_write = torch.zeros((_:=list(kv_cache.shape))).__setitem__(-2, x.shape[-1]) or _)
        ...