Python torch.flatten()函数案例详解

先看函数参数：torch.flatten(input,start_dim=0,end_dim=-1)input:一个tensor，即要被“推平”的tensor。start_dim:“推平”...