Last error is solved, but this time when I
print(output)

tensor([[0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893],
        [0.3079, 0.3893]], device='cuda:0', grad_fn=<AddmmBackward>)

Is this normal?