文件名称：火炬手：PyTorch层和神经网络的形状和尺寸推断（类似于Keras）
文件大小：1.26MB
文件格式：ZIP
更新时间：2024-02-24 06:41:07
machine-learning library framework deep-learning shape
版 文件 测验 覆盖范围 风格 聚酰亚胺 Python 火炬 码头工人 是基于库提供的自动形状和维度推理torch.nn在当前SOTA架构功能层+附加构建块（例如）。 上述无需用户干预（除了单呼torchlayers.build ）类似于一个在看到 。 主要功能： 大多数torch.nn模块的形状推断（卷积层，递归层，变压器层，注意力层和线性层） 尺寸推断（例如，基于input shape torchlayers.Conv用作torch.nn.Conv1d/2d/3d ） 自定义模块的形状推断（请参见示例部分） 其他torchlayers.Reshape图层（例如， torchlayers.Reshape或torchlayers.StandardNormalNoise ） 其他SOTA层主要来自ImageNet竞赛（例如 ， ， ） 有用的默认值（ "same"填充和Conv默认kernel_size=3 ，辍学率等） 零开销和支持 请记住，该库的工作几乎与最初的PyTorch完全一样。 这意味着您可以使用Sequential ，使用torch.nn.Module