torch.flatten. torch. flatten (input, start_dim=0, end_dim=- 1) → Tensor. Flattens input by reshaping it into a one-dimensional tensor.
確定! 回上一頁