cutest piggy 的博客

训练需要model.train() ：启用 BatchNormalization 和 Dropout测试必要(pytorch是默认开启BatchNormalization 和 Dropout的，测试或者pre-train的时候必须使用)model.eval() ：不启用 BatchNormalization 和 Dropout...

１ torch.cat用途用于将张量concatnate拼接，可按行、列拼接2 按行列拼接(注意张量维度)C = torch.cat( (A,B),0 ) #按维数0拼接（接着行竖着拼）C = torch.cat( (A,B),1 ) #按维数1拼接（接着列横着拼）示例import torchA=torch.ones(3,3)B=3*torch.ones(2,3)C=2*BD=...