start_dim用法
start_dim 是表示开始展开的维度
start_dim=0,就是从第一维度开始展开
start_dim=1 ,从第二维度开始展开,保留第一维度,第二维度改变
也可以 start_dim=-1 ,表示从倒数第二维度开始展开,那么倒数第二维度改变
end_dim用法
start_dim=0, end_dim=1,从第一维度开始展开,到第二维度结束,其他维度不变,以(3,2,2)维度看,那么便是 3,2 改变,最后的一维2保留,维度变成了 (6,2)

start_dim用法
start_dim 是表示开始展开的维度
start_dim=0,就是从第一维度开始展开
start_dim=1 ,从第二维度开始展开,保留第一维度,第二维度改变
也可以 start_dim=-1 ,表示从倒数第二维度开始展开,那么倒数第二维度改变
end_dim用法
start_dim=0, end_dim=1,从第一维度开始展开,到第二维度结束,其他维度不变,以(3,2,2)维度看,那么便是 3,2 改变,最后的一维2保留,维度变成了 (6,2)