Python chainer.utils.type_check 模块,prod() 实例源码
我们从Python开源项目中,提取了以下13个代码示例,用于说明如何使用chainer.utils.type_check.prod()。
def check_type_forward(self, in_types):
n_in = in_types.size()
type_check.expect(2 <= n_in, n_in <= 3)
x_type, w_type = in_types[:2]
type_check.expect(
x_type.dtype.kind == 'f',
w_type.dtype.kind == 'f',
x_type.ndim >= 2,
w_type.ndim == 2,
type_check.prod(x_type.shape[1:]) == w_type.shape[1],
)
if n_in.eval() == 3:
b_type = in_types[2]
type_check.expect(
b_type.dtype == x_type.dtype,
b_type.ndim == 1,
b_type.shape[0] == w_type.shape[0],
)
def check_type_forward(self, w_type = in_types[:2]
type_check.expect(
x_type.dtype == numpy.float32,
w_type.dtype == numpy.float32,
)
if n_in.eval() == 3:
b_type = in_types[2]
type_check.expect(
b_type.dtype == numpy.float32,
)
def check_type_forward(self, in_types):
type_check.expect(
in_types.size() == 1,
)
x_type, = in_types
cnt = _count_unkNown_dims(self.shape)
if cnt == 0:
type_check.expect(
type_check.prod(x_type.shape) == type_check.prod(self.shape))
else:
kNown_size = 1
for s in self.shape:
if s > 0:
kNown_size *= s
size_var = type_check.Variable(kNown_size,
'kNown_size(=%d)' % kNown_size)
type_check.expect(
type_check.prod(x_type.shape) % size_var == 0)
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",
g_type.ndim == 2,
)
if n_in.eval() == 4:
b_type = in_types[3]
type_check.expect(
b_type.dtype == x_type.dtype,
)
def check_type_forward(self, w_type = in_types[:2]
type_check.expect(
x_type.dtype == np.float32,
w_type.dtype == np.float32,
)
if n_in.eval() == 3:
b_type = in_types[2]
type_check.expect(
b_type.dtype == np.float32,
)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。