你即將離開本站

並前往https://stackoverflow.com/questions/68924374/use-torch-square-inside-a-nn-sequential-layer-in-pytorch

Use torch.square() inside a nn.Sequential layer in PyTorch

You can't put a PyTorch function in a nn.Sequential pipeline, it needs to be a nn.Module . You could wrap it like this: class Square(nn.

確定！ 回上一頁

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

  1. Torch sqrt
    2.
  2. Torch square 1.4 0
    3.
  3. Torch expand
    4.
  4. Pytorch any
    5.
  5. Torchtensor
    6.
  6. Torch slice
    7.
  7. Item pytorch
    8.
  8. Pytorch inner product
    9.