-
totensor 는
넘파이 h w c 에서 텐서 c h w 로 바뀌어 준다
범위도 0 에서 255에서 0에서1로 바꾸어준다cpu 가 load 하므로 배치사이즈가클경우 num worker 가 많을수록? 좋다
3채널을 입력으로 받아 64채널의 아웃풋
30*5*5의 이미지를 받아 flatten시켜서 받고 128 개의 fc 아웃풋
view 함수순번이 predicted 임
채널을 이용해 학습 인풋 weight 를 직접 넣어줘야 함
-1자동으로 숫자 선택 (16이 고정되어있는 상태)
3,3 대신 3,5 필터
argmax 는 max_indices이다