SSD network's permute layer. Args: input: A variable. Contains the feature map. Namely bottom[0] in caffe. dims: A vector. Contains the order. Returns: A variable. */
| 1388 | A variable. |
| 1389 | */ |
| 1390 | VARP _Permute(VARP input, INTS dims) { |
| 1391 | std::unique_ptr<OpT> op(new OpT); |
| 1392 | op->type = OpType_Permute; |
| 1393 | auto param = new PermuteT; |
| 1394 | param->dims = dims; |
| 1395 | op->main.type = OpParameter_Permute; |
| 1396 | op->main.value = param; |
| 1397 | return Variable::create(Expr::create(std::move(op), {input})); |
| 1398 | } |
| 1399 | /*SSD network's detectionoutput layer. |
| 1400 | Args: |
| 1401 | location: A variable. |