torch.flatten()
先看函数参数:torch.flatten(input, start_dim=0, end_dim=-1)input: 一个 tensor,即要被“推平”的 tensor。start_dim: “推平”的起始维度。end_dim:“推平”的结束维度。首先如果按照 start_dim 和 end_dim 的默认值,那么这个函数会把 input 推平成一个 shape 为 ...