提交 5e216fcf 编写于 作者: T tink2123

fix split

test=develop
上级 e296e0fe
......@@ -4883,7 +4883,7 @@ def split(input, num_or_sections, dim=-1, name=None):
assert num_or_sections > 1, 'num_or_sections must be more than 1.'
num = num_or_sections
else:
assert len(num_or_sections) < input_shape[
assert len(num_or_sections) <= input_shape[
dim], 'len(num_or_sections) must not be more than input.shape[dim].'
num = len(num_or_sections)
outs = [
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册