torch.flatten(input, start_dim=0, end_dim=-1). input: A tensor, that is, the tensor to be "flattened". start_dim: The starting dimension of ...
確定! 回上一頁