提交 80aadc8f authored 作者: Iulian Vlad Serban's avatar Iulian Vlad Serban

Another minor modification to #2613.

上级 3ec221fe
...@@ -3058,13 +3058,9 @@ class GpuJoin(tensor.Join, GpuOp): ...@@ -3058,13 +3058,9 @@ class GpuJoin(tensor.Join, GpuOp):
axis_int = axis axis_int = axis
if (axis_int < 0): if (axis_int < 0):
# Find tensor with smallest number of dimensions # Since all tensors must have the same number of dimensions,
min_dim = -1 # we simply add the number of dimensions for the first tensor
for cnda in tensors: axis = axis + as_tensor_variable_args[0].ndim
if min_dim < 0 or min_dim > cnda.ndim:
min_dim = cnda.ndim
axis = axis + min_dim
output_maker = \ output_maker = \
lambda bcast: CudaNdarrayType(broadcastable=bcast)() lambda bcast: CudaNdarrayType(broadcastable=bcast)()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论