Crate caffe2op_lengthstop
source ·Structs
- | Apply TopK to each segment of the input | tensor, where segments are defined | by their LENGTHS, and concatenate them | in an output tensor of shape=(SIZE(LENGTHs), | k). | | In case there’s less than k values in | a segment, the output value will be padded | by 0, and the corresponding output indices | will be padded by -1. |