pytorch 中的reshape()、view()、transpose()和flatten()_a1367666195的博客-程序员秘密 ... torch.flatten(input, start_dim=0, end_dim=-1) → Tensor.
確定! 回上一頁