提交 884b7d67 authored 作者: Florian Bordes's avatar Florian Bordes

Raising error if axis type isn't valid

上级 fa3e9b92
......@@ -1003,7 +1003,7 @@ def check_and_normalize_axes(x, axis):
elif isinstance(axis.data, (list, np.ndarray)):
axis = [int(i) for i in axis.data]
else:
axis = []
raise TypeError("Axis must be an integer, tuple, list of integers or a TensorVariable. Got %s" % axis)
if len(axis) > 0:
for i in range(len(axis)):
if axis[i] < 0:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论