先看函数参数: torch.flatten(input, start_dim=0, end_dim=-1) input: 一个tensor,即要被“推平”的tensor. start_dim: “推平”的起始维度. end_dim: “推平”的结束维度.
確定! 回上一頁