赞
踩
torch.zeros:用来将tensor中元素值全置为0
a=torch.zeros(2,2)
''
tensor([[0., 0.],
[0., 0.]])
''
b=torch.zeros(3)
''
tensor([0., 0., 0.])
''
torch.ones:用来将tensor中元素值全置为1
c=torch.ones(2,2)
''
tensor([[1., 1.],
[1., 1.]])
''
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。