weixin_50456250的博客

1.torch.randn(3,4)是随机生成 ，三行四列的矩阵 2.x.ge(0.5)是找出比0.5还大的数字 3.torch.masked_select(x,mask)是将大于0.5的数字列出来 4. .shape就是将所有的那些形状打出来 1、a.shape 指的是各个维度的模型数字，比如三行六列这样 2、torch.Size[(4,3,28,28)],从右到左，维度依次增大 3、a[...].shape是将所有维度打出 4、a[0,...].shape是将所后面三个的维度打..