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是取>的意思,