pytorch 记录
item()把张量中的元素变成数字,比直接用索引精度要高一些
torch.unsqueeze()可以拓展一维 squeeze可以去掉一维(这两者只对维数为1的轴生效)
torch.expand 可以对张量进行扩张,但注意,只能扩张维数为1或0的轴,如下图,对于维数为2的轴进行扩张就报错了
torch.transpose()和torch.permute()可以对轴进行变换,transpose是调换,permute是按照新的顺序重构。
torch.cat(inputs, dimension=0) :在给定轴上对输入的张量进行拼接
torch.stack():沿着一个新维度对输入张量序列进行拼接
张量的索引
掩码取值
其中ge是取>的意思,