xmy21sunshine的博客

复杂度

广播机制

import redata1_loss=[]data2_loss=[]#文件中各提取关键数值所在的位置Epoch = 0train_loss = 0train_acc = 1test_loss =2test_acc = 3f =open(“./loss.txt”, ‘r’, encoding=“utf8”)lines = f.readline()while lines:result1 = re.findall(r’\d+',lines) # 读取整数result2 = re.fi

torch.cat 将两个tensor拼接在一起，即拼接、联系在一起的意思；使用torch.cat((A, B),dim)时，除拼接dim维对应的值不同外，其余维的数值需一致A = torch.ones(2,3)B = 2 * torch.ones(4,3)C = torch.cat((A, B),dim=0)print(C.shape)D = 2 * torch.ones(2,4)C = torch.cat((A, D),dim=1)print(C.shape)...