linear.py 文件源码

python
阅读 33 收藏 0 点赞 0 评论 0

项目:ddnn 作者: kunglab 项目源码 文件源码
def check_type_forward(self, in_types):
        n_in = in_types.size()
        type_check.expect(3 <= n_in, n_in <= 4)
        x_type, w_type, g_type = in_types[:3]

        type_check.expect(
            x_type.dtype.kind == "f",
            w_type.dtype.kind == "f",
            g_type.dtype.kind == "f",
            x_type.ndim >= 2,
            w_type.ndim == 2,
            g_type.ndim == 2,
            type_check.prod(x_type.shape[1:]) == w_type.shape[1],
        )

        if n_in.eval() == 4:
            b_type = in_types[3]
            type_check.expect(
                b_type.dtype == x_type.dtype,
                b_type.ndim == 1,
                b_type.shape[0] == w_type.shape[0],
            )
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号