torch.flatten(input, start_dim=0, end_dim=-1) → Tensor. >>> t = torch.tensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]]]) >>> torch.flatten(t) ...
確定! 回上一頁