提交 b6eb49b3 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Add comment

上级 886dc49a
......@@ -3871,6 +3871,9 @@ class Join(Op):
# The joining dimension could be negative, but we need it to be
# in [0, n_dim) in the loop below.
# An axis < -n_dim or >= ndim would be invalid, but this is
# not checked here. An Assert op would be a way of addressing that,
# but it may disrupt optimizations.
join_dim = switch(ge(node.inputs[0], 0),
node.inputs[0],
node.inputs[0] + n_dim)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论