This could be achieved by calling view on the output of the last nn.MaxPool2d, but unfortunately, we don't have any ... Flatten starting from PyTorch 1.3.
確定! 回上一頁