torch.Size([4])
tensor([0.5313, 1.3607, 0.7891, 1.3110])