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