https://blog.pscpeng.xyz:4433/archives/pytorch-construct-tensor