火炬手:PyTorch层和神经网络的形状和尺寸推断(类似于Keras) 源码

版 文件 测验 覆盖范围 风格 聚酰亚胺 Python 火炬 码头工人 是基于库提供的自动形状和维度推理torch.nn在当前SOTA架构功能层+附加构建块(例如)。 上述无需用户干预(除了单呼torchlayers.build )类似于一个在看到 。 主要功能: 大多数torch.nn模块的形状推断(卷积层,递归层,变压器层,注意力层和线性层) 尺寸推断(例如,基于input shape torchlayers.Conv用作torch.nn.Conv1d/2d/3d ) 自定义模块的形状推断(请参见示例部分) 其他torchlayers.Reshape图层(例如, torchla