[Pytorch] torchscript의 List
torchscript는 list를 생성하게 되면 기본적으로 List[T], 즉, tensor를 member로 가지는 list가 생성이 된다. 그냥 python code를 실행할 때는 문제가 되지 않는 코드이다. tmp = [] for i in range(0, 10, 1): tmp.append(i) new_tmp = torch.tensor(tmp) ==> 빈 list를 하나 선언해주고 그 안에 int형을 담은 다음 list를 tensor로 변환해주는 torch.tensor 함수를 사용. 그러나 이 부분을 torchscript로 변환하려고 하면 error가 뜬다. 잘 변환이 되도 C++에서 불러와 쓰려고 하면 저 부분에서 다음과 같은 runtime error가 뜬다. RuntimeError: Input mu..
이것저것 자료/Python
2021. 10. 6. 13:06
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 코딩테스트
- shellscript
- CUDA
- BFS
- 설치하기
- dfs
- 백준
- notfound
- pytorch
- Python
- 동적프로그래밍
- 최소신장트리
- torch
- 백트래킹
- version
- 프로그래머스
- numpy
- LGSVL
- torchscript
- 파이썬
- tensorflow
- 설치
- PIP
- docker
- 이것이코딩테스트다
- n과m
- matplotlib
- 다익스트라
- 카카오
- error
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 1 | 2 | 3 | ||||
| 4 | 5 | 6 | 7 | 8 | 9 | 10 |
| 11 | 12 | 13 | 14 | 15 | 16 | 17 |
| 18 | 19 | 20 | 21 | 22 | 23 | 24 |
| 25 | 26 | 27 | 28 | 29 | 30 | 31 |
글 보관함