你即將離開本站

並前往http://www.4k8k.xyz/article/weixin_45281949/103282148

torch.sum(),dim=0,dim=1解析_倔强的春苗的博客-程序员资料

b = torch.sum(a,dim=0) print(b) b.shape. output: tensor([ 5, 7, 9]) torch.Size([3]). 5 = 1 + 4 7 = 2 + 5 9 = 3 + 6. 此时b为1维张量，是一个向量，且剩余元素个 ...

確定！ 回上一頁

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