torch.zeros() and torch.ones()

import torch

if __name__ == '__main__':

    a = torch.zeros(3,2)
    print(a)
    b = torch.ones(12)
    print(b)
    

result：

tensor([[0., 0.],
        [0., 0.],
        [0., 0.]])
tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.])