def METHOD_NAME(self, index: slice, value) -> None:
    if not (index.start is index.stop is index.step is None):  # noqa: E714
        raise NotImplementedError("only slice [:] supported")
    if not self.atomic:
        self._values = value
        return
    if len(value) != 1:
        raise IndexError("atomic batch cannot be replaced with multiple tensors")
    self._values = value[0]