你即將離開本站

並前往https://github.com/pytorch/pytorch/issues/67670

[ONNX] torch.flatten for 1D tensor results in wrong ONNX

Bug It seems if you export torch.flatten(x) where x.ndim == 1, torch.onnx exports a single Flatten op whose output is always 2D.

確定！ 回上一頁

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

  1. Tensor flatten
    2.
  2. Flatten parameters pytorch
    3.
  3. Pytorch flatten view
    4.
  4. Numpy flatten
    5.
  5. Nn Sequential squeeze
    6.
  6. Torch view
    7.
  7. Pytorch Dense
    8.
  8. Torch reshape
    9.