当前位置:   article > 正文

torch.zeros_like()和torch.ones_like()简要介绍_torch.tenso.like

torch.tenso.like

定义

  • torch.zeros_like(input)
  • torch.ones_like(input)

用法

分别生成和输入数据维度一致的全为0或全为1的内容

示例

  1. import torch
  2. a = torch.tensor([[1,2,3],[4,5,6]])
  3. print(torch.zeros_like(a))
  4. print(torch.ones_like(a))
  5. >>>tensor([[0, 0, 0],
  6. [0, 0, 0]])
  7. tensor([[1, 1, 1],
  8. [1, 1, 1]])

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

闽ICP备14008679号