Method join

caffe2/python/operator_test/partition_ops_test.py:71–74

(a)

                                accum.append(data[j])

                        def join(a):
                            if not a:
                                return np.empty(shape=(0, ) + suffix_shape)
                            return np.stack(a)

                        out.append(join(accum))
                return out