os.path.basename(filename)
返回filename的文件名
os.path.splittext(filename)
返回文件名和扩展名
torch.cat(A,B,dim=0)
将矩阵A,B按行拼接起来,若dim=1,则按列拼接
torch.rand(4,1,3).squeeze()
去除维度为1 运行结果维度为4x3
os.path.basename(filename)
返回filename的文件名
os.path.splittext(filename)
返回文件名和扩展名
torch.cat(A,B,dim=0)
将矩阵A,B按行拼接起来,若dim=1,则按列拼接
torch.rand(4,1,3).squeeze()
去除维度为1 运行结果维度为4x3