torch. flatten (input, start_dim=0, end_dim=-1) → Tensor¶. Flattens a contiguous range of dims in a tensor. Parameters. input (Tensor) – the input tensor.
確定! 回上一頁