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.]])
承接Matlab、Python和C++的编程,机器学习、计算机视觉的理论实现及辅导,本科和硕士的均可,咸鱼交易,专业回答请走知乎,详谈请联系QQ号757160542,非诚勿扰。