你即將離開本站

並前往https://ithelp.ithome.com.tw/articles/10243145

MNIST 手寫數字辨識 - iT 邦幫忙

... F.relu(x) x = F.max_pool2d(x, 2) x = self.dropout(x) x = torch.flatten(x, ... 要predict），所以用flatten 把feature 集中成vector 後，再用fully-connected ...

確定！ 回上一頁

查詢 「Torch flatten」的人也找了：

  1. Torch nn flatten
    2.
  2. Pytorch flatten view
    3.
  3. Torch view
    4.
  4. Torch reshape
    5.
  5. Torchcat
    6.
  6. Torch squeeze
    7.
  7. Numpy flatten
    8.
  8. Torch expand
    9.