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