인프런 커뮤니티 질문&답변
작성한 질문수
output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]
작성
·
655
답변 1
0
코코
지식공유자
안녕하세요. 님
답변이 늦어서 죄송합니다. DCGAN에 대한 에러도 질문을 해주셨는데 여기서 답변을 같이 해드리도록 하겠습니다.
우선 제가 올려드린 코드는 다시 확인해 보았는데 정상적으로 잘 작동하는 것을 확인하였습니다.
올려주신 캡쳐화면으로는 원인을 알기는 쉽지 않은 것 같은데
추정하기로는 저와 환경이 달라서 그런게 아닌가 싶습니다.
이전에 질문을 올려주신바와 같이 Type error는
에러 메세지에 따라서 다음과 같이 수정해 주시면 됩니다.
output.type(torch.LongTensor)
output.type(torch.FloatTensor)
이에 관해서는 아래링크를 참고해주시면 감사하겠습니다.
https://stackoverflow.com/questions/60440292/runtimeerror-expected-scalar-type-long-but-found-float
더불어 제 Python 환경은 3.6.5이고 Torch는 1.0.0 입니다.