当前位置:   article > 正文

pytorch中的mean,max操作_x.mean(1, keepdim=true).detach()

x.mean(1, keepdim=true).detach()
  1. x=torch.arange(15).view(5,3)

  2. x=x.float()

  3. x_mean=torch.mean(x,dim=0,keepdim=True)(表示每一列的平均数)

  4. x_mean0=torch.mean(x,dim=1,keepdim=True)(表示每一行的平均数)

  5. x_mean:6 7 8

  6.  
  7. x_mean0:

1

4

7

10

13

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/606281
推荐阅读
相关标签
  

闽ICP备14008679号