| 112 | } |
| 113 | |
| 114 | Tensor Stack(std::vector<Tensor> &tensorList, std::optional<Stream> pstream) |
| 115 | { |
| 116 | int64_t outputShape[4] = {}; // NCHW/NHWC |
| 117 | nvcv::TensorLayout layout = nvcv::TENSOR_CHW; |
| 118 | nvcv::DataType dtype; |
| 119 | checkTensorList(tensorList, outputShape, layout, dtype); |
| 120 | |
| 121 | //create new output tensor |
| 122 | Tensor output = Tensor::Create( |
| 123 | { |
| 124 | {outputShape[0], outputShape[1], outputShape[2], outputShape[3]}, |
| 125 | layout |
| 126 | }, |
| 127 | dtype); |
| 128 | return StackIntoInternal(output, tensorList, pstream, outputShape[0]); |
| 129 | } |
| 130 | |
| 131 | } // namespace |
| 132 |
nothing calls this directly
no test coverage detected