RuntimeError:invalid argument 2:size ‘[1x18x256x256]‘ is invalid for input with 196608 elements as

    技术2022-07-11  82

    发现是用了view之后导致的错误

    本来view之后的1x18x256x256 = 1179648 ,但和196608不符合, view的形状可能有问题

    Processed: 0.008, SQL: 9