반응형 numpy2 [에러] TypeError : can't convert CUDA tensor to numpy 에러 원인 : GPU에 할당되어 있는 tensor를 Numpy 배열로 변환할 때 생기는 에러 gpu에 할당된 tensor를 .numpy()로 변경을 하려고 할 때 생김 .cpu( )를 사용하여 gpu에 할당된 tensor를 cpu tensor로 변환 # if X is CPU tensor X.cpu().numpy() 2023. 2. 9. [numpy] numpy 소수점 반올림하기 / array 전체 출력 / 출력 형식 변경 반올림을 하거나 소수점 제한을 줘서 출력할 때, numpy에 있는 np.set_printoptions라는 메소드라는 것을 사용하면 된다. jupyter로 작업을 할 경우, 상단의 셀에서 한번만 설정해주면 해당 파일의 작업들 전체에 설정을 따르게 된다. 소수점 반올림하기(precision) * default = 8 np.array([0.123456789]) array([0.12345679]) np.set_printoptions(precision=5) np.array([0.123456789]) array([0.12346]) np.set_printoptions(precision=8) np.array([0.123456789]) array([0.12345679]) array 전체 출력(threshold) * de.. 2023. 1. 25. 이전 1 다음 반응형