torch.zeros_like
torch.zeros_like:生成和括号内变量维度维度一致的全是零的内容。
import torch
a = torch.rand(5,1)
print(a)
n=torch.zeros_like(a)
print('n=',n)
tensor([[0.9653],
[0.5581],
[0.1648],
[0.3715],
[0.2194]])
n= tensor([[0.],
[0.],
[0.],
[0.],
[0.]])本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2021-03-26 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读