| ## @package arra_helpers | |
| # Module caffe2.python.helpers.array_helpers | |
| def concat(model, blobs_in, blob_out, **kwargs): | |
| """Depth Concat.""" | |
| if kwargs.get('order') and kwargs.get('axis'): | |
| # The backend throws an error if both are given | |
| kwargs.pop('order') | |
| return model.net.Concat( | |
| blobs_in, | |
| [blob_out, "_" + blob_out + "_concat_dims"], | |
| **kwargs | |
| )[0] | |
| def depth_concat(model, blobs_in, blob_out, **kwargs): | |
| """The old depth concat function - we should move to use concat.""" | |
| print("DepthConcat is deprecated. use Concat instead.") | |
| return concat(blobs_in, blob_out, **kwargs) | |