4f58fc9c82
* Move utils classes into models * Add deprecation warnings * Remove from docs * Update config attributes check
2.1 KiB
カスタムレイヤーとユーティリティ
このページには、ライブラリで使用されるすべてのカスタム レイヤーと、モデリングに提供されるユーティリティ関数がリストされます。
これらのほとんどは、ライブラリ内のモデルのコードを研究する場合にのみ役に立ちます。
Pytorch custom modules
autodoc pytorch_utils.Conv1D
PyTorch Helper Functions
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 custom layers
autodoc modeling_tf_utils.TFConv1D
autodoc modeling_tf_utils.TFSequenceSummary
TensorFlow loss functions
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 Helper Functions
autodoc modeling_tf_utils.get_initializer
autodoc modeling_tf_utils.keras_serializable
autodoc modeling_tf_utils.shape_list