2.2 KiB
自定义层和工具
此页面列出了库使用的所有自定义层，以及它为模型提供的实用函数。
其中大多数只有在您研究库中模型的代码时才有用。
Pytorch自定义模块
autodoc pytorch_utils.Conv1D
autodoc modeling_utils.PoolerStartLogits - forward
autodoc modeling_utils.PoolerEndLogits - forward
autodoc modeling_utils.PoolerAnswerClass - forward
autodoc modeling_utils.SquadHeadOutput
autodoc modeling_utils.SQuADHead - forward
autodoc modeling_utils.SequenceSummary - forward
PyTorch帮助函数
autodoc pytorch_utils.apply_chunking_to_forward
autodoc pytorch_utils.find_pruneable_heads_and_indices
autodoc pytorch_utils.prune_layer
autodoc pytorch_utils.prune_conv1d_layer
autodoc pytorch_utils.prune_linear_layer
TensorFlow自定义层
autodoc modeling_tf_utils.TFConv1D
autodoc modeling_tf_utils.TFSequenceSummary
TensorFlow loss 函数
autodoc modeling_tf_utils.TFCausalLanguageModelingLoss
autodoc modeling_tf_utils.TFMaskedLanguageModelingLoss
autodoc modeling_tf_utils.TFMultipleChoiceLoss
autodoc modeling_tf_utils.TFQuestionAnsweringLoss
autodoc modeling_tf_utils.TFSequenceClassificationLoss
autodoc modeling_tf_utils.TFTokenClassificationLoss
TensorFlow帮助函数
autodoc modeling_tf_utils.get_initializer
autodoc modeling_tf_utils.keras_serializable
autodoc modeling_tf_utils.shape_list