提交 016e474e authored 作者: Harm de Vries's avatar Harm de Vries

flake8

上级 f6f18c88
......@@ -1363,8 +1363,8 @@ class MaxAndArgmax(Op):
keep_axes = numpy.array([i for i in range(x.ndim) if i not in axes])
# Not-reduced axes in front
transposed_x = numpy.transpose(x, numpy.concatenate((keep_axes, axes)))
reshaped_x = transposed_x.reshape(
transposed_x.shape[:len(keep_axes)] + (-1,))
reshaped_x = transposed_x.reshape(transposed_x.shape[:len(keep_axes)] +
(-1,))
max_idx[0] = theano._asarray(numpy.argmax(reshaped_x, axis=-1),
dtype='int64')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论