Fix IndexError

✅ Solution

IndexError in PyTorch usually arises when accessing a tensor element using an invalid index (out of bounds). To fix it, ensure that all indices used to access tensor elements are within the valid range (0 to tensor.size(dimension) - 1) for each dimension; use `assert` statements to help debug problematic indices.