您好,欢迎光临本网站![请登录][注册会员]  

搜索资源列表

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

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

    • 发布日期:2021-02-03
    • 文件大小:1048576
    • 提供者:weixin_42122878