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