参数

• raw_speech (torch.Tensor, np.ndarray, List[float], List[torch.Tensor], List[np.ndarray], List[List[float]]) — 要填充的序列或序列批次。每个序列可以是张量、numpy 数组、浮点值列表、张量列表、numpy 数组列表或浮点值列表的列表。必须是单声道音频，而不是立体声，即每个时间步单个浮点数。
• padding (bool, str 或 PaddingStrategy, 可选, 默认为 False) — 选择一种策略来填充返回的序列（根据模型的填充侧和填充索引）：
  • True 或 'longest'：填充到批次中最长的序列（如果仅提供单个序列，则不填充）。
  • 'max_length'：填充到使用参数 max_length 指定的最大长度，或者如果未提供该参数，则填充到模型可接受的最大输入长度。
  • False 或 'do_not_pad' (默认)：不填充（即，可以输出具有不同长度序列的批次）。
• max_length (int, 可选) — 返回列表的最大长度，以及可选的填充长度（见上文）。
• truncation (bool) — 激活截断，将长于 max_length 的输入序列剪切为 max_length。
• pad_to_multiple_of (int, optional) — 如果设置，将序列填充为提供值的倍数。

  这对于在计算能力 >= 7.5 (Volta) 的 NVIDIA 硬件上或在受益于序列长度为 128 的倍数的 TPU 上启用 Tensor Cores 的使用尤其有用。

• return_attention_mask (bool, optional) — 是否返回 attention mask。 如果保留为默认值，将根据特定 feature_extractor 的默认值返回 attention mask。

  什么是 attention masks？

• return_tensors (str 或 TensorType, optional) — 如果设置，将返回 tensors 而不是 python 整数列表。可接受的值为：
  • 'tf': 返回 TensorFlow tf.constant 对象。
  • 'pt': 返回 PyTorch torch.Tensor 对象。
  • 'np': 返回 Numpy np.ndarray 对象。
• sampling_rate (int, optional) — raw_speech 输入被采样的采样率。 强烈建议在前向调用时传递 sampling_rate 以防止静默错误。
• padding_value (float, defaults to 0.0) —

参数

• pretrained_model_name_or_path (str 或 os.PathLike) — 这可以是以下之一：
  • 一个字符串，托管在 huggingface.co 上的模型仓库内的预训练 feature_extractor 的模型 ID。
  • 一个指向目录的路径，该目录包含使用 save_pretrained() 方法保存的 feature extractor 文件，例如，./my_model_directory/。
  • 指向已保存的 feature extractor JSON 文件的路径或 URL，例如，./my_model_directory/preprocessor_config.json。 **kwargs — 传递给 from_pretrained() 和 ~tokenization_utils_base.PreTrainedTokenizer.from_pretrained 的其他关键字参数。

参数

• save_directory (str 或 os.PathLike) — 将在其中保存 feature extractor JSON 文件和 tokenizer 文件的目录（如果目录不存在，将创建目录）。
• push_to_hub (bool, optional, defaults to False) — 保存后是否将您的模型推送到 Hugging Face 模型中心。 您可以使用 repo_id 指定要推送到的存储库（默认为您的命名空间中 save_directory 的名称）。
• kwargs (Dict[str, Any], optional) — 传递给 push_to_hub() 方法的其他关键字参数。

参数

• input_features (形状为 (batch_size, sequence_length) 的 torch.LongTensor) — 词汇表中输入序列 tokens 的索引。

  索引可以使用 Wav2Vec2CTCTokenizer 获得。 详情请参阅 PreTrainedTokenizer.encode() 和 PreTrainedTokenizer.call()。

  什么是输入 IDs？

• attention_mask (形状为 (batch_size, sequence_length) 的 torch.FloatTensor, 可选) — 用于避免在 padding token 索引上执行 attention 的掩码。 掩码值在 [0, 1] 中选择：
  • 1 表示 token 未被掩盖，
  • 0 表示 token 已被掩盖。

  什么是 attention 掩码？

• head_mask (形状为 (num_heads,) 或 (num_layers, num_heads) 的 torch.FloatTensor, 可选) — 用于 nullify self-attention 模块中选定 head 的掩码。 掩码值在 [0, 1] 中选择：
  • 1 表示 head 未被掩盖，
  • 0 表示 head 已被掩盖。
• output_attentions (bool, 可选) — 是否返回所有 attention 层的 attentions tensors。 有关更多详细信息，请参阅返回的 tensors 下的 attentions。
• output_hidden_states (bool, 可选) — 是否返回所有层的 hidden states。 有关更多详细信息，请参阅返回的 tensors 下的 hidden_states。
• return_dict (bool, 可选) — 是否返回 ModelOutput 而不是普通的 tuple。

一个 transformers.modeling_outputs.BaseModelOutput 或一个 torch.FloatTensor 的 tuple (如果传递了 return_dict=False 或者当 config.return_dict=False 时)，其中包含各种元素，具体取决于配置 (MCTCTConfig) 和输入。

• last_hidden_state (形状为 (batch_size, sequence_length, hidden_size) 的 torch.FloatTensor) — 模型最后一层的输出端的 hidden-states 序列。

• hidden_states (tuple(torch.FloatTensor), 可选, 当传递了 output_hidden_states=True 或当 config.output_hidden_states=True 时返回) — torch.FloatTensor 的 Tuple (对于 embeddings 的输出，如果模型有 embedding 层，则有一个；对于每一层的输出，则有一个)，形状为 (batch_size, sequence_length, hidden_size)。

  模型在每一层输出端的 Hidden-states，加上可选的初始 embedding 输出。

• attentions (tuple(torch.FloatTensor), 可选, 当传递了 output_attentions=True 或当 config.output_attentions=True 时返回) — torch.FloatTensor 的 Tuple (每层一个)，形状为 (batch_size, num_heads, sequence_length, sequence_length)。

  attention softmax 之后的 Attention 权重，用于计算 self-attention heads 中的加权平均值。

参数

• input_features (形状为 ({0}) 的 torch.LongTensor) — 词汇表中输入序列 tokens 的索引。

  索引可以使用 Wav2Vec2CTCTokenizer 获得。 详情请参阅 PreTrainedTokenizer.encode() 和 PreTrainedTokenizer.call()。

  什么是输入 IDs？

• attention_mask (形状为 ({0}) 的 torch.FloatTensor, 可选) — 用于避免在 padding token 索引上执行 attention 的掩码。 掩码值在 [0, 1] 中选择：
  • 1 表示 token 未被掩盖，
  • 0 表示 token 已被掩盖。

  什么是 attention 掩码？

• head_mask (形状为 (num_heads,) 或 (num_layers, num_heads) 的 torch.FloatTensor, 可选) — 用于 nullify self-attention 模块中选定 head 的掩码。 掩码值在 [0, 1] 中选择：
  • 1 表示 head 未被掩盖，
  • 0 表示 head 已被掩盖。
• output_attentions (bool, 可选) — 是否返回所有 attention 层的 attentions tensors。 有关更多详细信息，请参阅返回的 tensors 下的 attentions。
• output_hidden_states (bool, 可选) — 是否返回所有层的 hidden states。 有关更多详细信息，请参阅返回的 tensors 下的 hidden_states。
• return_dict (bool, 可选) — 是否返回 ModelOutput 而不是普通的 tuple。
• labels (形状为 (batch_size, target_length) 的 torch.LongTensor, 可选) — 用于 connectionist temporal classification 的标签。 请注意，target_length 必须小于或等于输出 logits 的序列长度。 索引在 [-100, 0, ..., config.vocab_size - 1] 中选择。 所有设置为 -100 的标签都会被忽略（掩盖），损失仅针对 [0, ..., config.vocab_size - 1] 中的标签计算。

一个 transformers.modeling_outputs.CausalLMOutput 或一个 torch.FloatTensor 的 tuple (如果传递了 return_dict=False 或者当 config.return_dict=False 时)，其中包含各种元素，具体取决于配置 (MCTCTConfig) 和输入。

• loss (形状为 (1,) 的 torch.FloatTensor, 可选, 当提供 labels 时返回) — 语言建模损失（用于下一个 token 预测）。

• logits (形状为 (batch_size, sequence_length, config.vocab_size) 的 torch.FloatTensor) — 语言建模 head 的预测分数（SoftMax 之前每个词汇表 token 的分数）。

• hidden_states (tuple(torch.FloatTensor), 可选, 当传递了 output_hidden_states=True 或当 config.output_hidden_states=True 时返回) — torch.FloatTensor 的 Tuple (对于 embeddings 的输出，如果模型有 embedding 层，则有一个；对于每一层的输出，则有一个)，形状为 (batch_size, sequence_length, hidden_size)。

  模型在每一层输出端的 Hidden-states，加上可选的初始 embedding 输出。

• attentions (tuple(torch.FloatTensor), 可选, 当传递了 output_attentions=True 或当 config.output_attentions=True 时返回) — torch.FloatTensor 的 Tuple (每层一个)，形状为 (batch_size, num_heads, sequence_length, sequence_length)。

  attention softmax 之后的 Attention 权重，用于计算 self-attention heads 中的加权平均值。