你即將離開本站

並前往https://www.coder.work/article/6189884

python - 在torch.sum() 中dim=-1 或-2 是什么意思？ - IT工具网

让我以一个二维矩阵为例: mat = torch.arange(9).view(3, -1) tensor([[0, 1, 2], [3, 4, 5], [6, 7, 8]]) torch.sum(mat, dim=-2) tensor([ 9, 12, 15])

確定！ 回上一頁

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

  1. Torch add
    2.
  2. Torch unsqueeze
    3.
  3. Torchtensor
    4.
  4. Torch square
    5.
  5. Torch divide
    6.
  6. Torchcat
    7.
  7. Torch expand
    8.
  8. Torch mean list
    9.