这个屌丝很懒，什么也没留下！

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

作者：从前慢现在也慢 | 2024-07-16 03:26:29

torch.tenso.like

定义

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

用法

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

示例


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

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