docs/source/zh/internal/modeling_utils.md

自定义层和工具

此页面列出了库使用的所有自定义层，以及它为模型提供的实用函数。

其中大多数只有在您研究库中模型的代码时才有用。

Pytorch自定义模块

[[autodoc]] pytorch_utils.Conv1D

PyTorch帮助函数

[[autodoc]] pytorch_utils.apply_chunking_to_forward

[[autodoc]] pytorch_utils.prune_linear_layer