参考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.]])
这篇博客介绍了PyTorch中torch.zeros_like函数的用法,通过示例展示了如何根据已有张量创建形状相同的全零张量。在代码中,a是一个随机生成的5x1维度张量,n则利用torch.zeros_like(a)生成了一个与a维度一致的全零张量。
385

被折叠的 条评论
为什么被折叠?



