You can use something like this . x.view(-1, size of the convolution Example :32*32*64). You can check for a full example here.
確定! 回上一頁