Crate caffe2op_concatsplit

Structs

ConcatOp
| Concatenate a list of tensors into a | single tensor. Similar functionality | to | | Numpy’s concatenate | function. The axis argument specifies | what axis along which the arrays will | be concatenated. | | When set to non-zero (default=0), the | add_axis argument adds the axis specified | in axis to all input tensors. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/concat_split_op.cc | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/concat_split_op.h |
GetConcatGradient
GetSplitGradient
SplitByLengthsOp
| Split a tensor into a list of tensors, | given a lengths input, along the specified | ‘axis’. If K outputs are provided, | the op assumes len(lengths) % K == 0. | | The input will be split into K parts. | Each part of length sum(lengths[i*k:i*k+k)) |
SplitOp
| Split an input tensor into a list of | tensors, along the axis specified by | the axis dimension. The lengths of | the split can be specified using argument | split or optional second input blob | to the operator. Otherwise, the tensor | is split to equal sized parts. | | Github Links: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/concat_split_op.cc |