torch.zeros和torch.ones

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.]])
''