Function Reformat

python/mod_cvcuda/operators/OpReformat.cpp:48–55

}

Tensor Reformat(Tensor &input, const nvcv::TensorLayout &out_layout, std::optional<Stream> pstream)
{
    nvcv::TensorShape out_shape = Permute(input.shape(), out_layout);

    Tensor output = Tensor::Create(out_shape, input.dtype());

    return ReformatInto(output, input, pstream);
}

} // namespace