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],
)
评论列表
文章目录