你即將離開本站

並前往https://coderedirect.com/questions/581135/what-does-dim-1-or-2-mean-in-torch-sum

what does dim=-1 or -2 mean in torch.sum()? - Code Redirect

let me take a 2D matrix as example: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, ...

確定！ 回上一頁

查詢 「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.