로딩
요청 처리 중입니다...

[torch] torch.sparse_coo_tensor 알아보기 feat. Coordinate format (COO format)

 [torch] torch.sparse_coo_tensor 알아보기 feat. Coordinate format (COO format)

우연히 open source code를 보다가 torch.sparse.FloatTensor 를 보게 되었는데 도통 검색해도 무슨 뜻인지 이해하기 어려웠다. 다행이도 구글에는 없는게 없었고, 겨우겨우 이해하게 되었는데, 미래의 나를 위하여 정리한다. torch를 통해 sparse를 만들때 torch.sparse_coo_tensor 를 사용하게 된다. torch.sparse.FloatTensor는 나중에 사라질 수도 있다고 적혀있으니 sparse_coo_tensor를 사용하자 (사실 둘다 하는 역할과 결과는 같다) Pytorch의 공식 문서에 따르면 torch.sparse.FloatTensor / torch.sparse_coo_tensor 의 정의는 다음과 같다.

PyTorch implements the so-called Coordinate format, or COO format, as one of the storage formats for implementing sparse tenso...

# COO # format # sparse # sparse_coo_tensor # torch