pytorch 회전 onnx 주의사항 (번역)

데이터 [index] = new데이터와 같은 장량은 현재 내보내기에서 지원되지 않습니다.이런 문제를 해결하는 방법의 하나는 산자 산점을 사용하여 원시 장량을 현저하게 갱신하는 것이다.바로 tensorflow의 정적 그림처럼 tensor의 값을 마음대로 바꿀 수 없으며 torch의scatter오류 해결 방법

# def forward(self, data, index, new_data):
#     data[index] = new_data          #  
#     return data

올바른 방식

class InPlaceIndexedAssignmentONNX(torch.nn.Module):
    def forward(self, data, index, new_data):
        new_data = new_data.unsqueeze(0)
        index = index.expand(1, new_data.size(1))
        data.scatter_(0, index, new_data)
        return data

LSTM과 같은 동적 변수의 클래스를 장착