pytorch 中的reshape()、view()和flatten() ... torch.flatten()的輸入是tensor torch.flatten(input, start_dim=0, end_dim=-1) → Tensor.
確定! 回上一頁