blace.ai
|
#include <public_ops.h>
Public Member Functions | |
IndexOp (blace::ops::OpP input, ml_core::BlaceIndexVec indices, std::optional< ml_core::ColorFormatEnum > color_format=std::nullopt) | |
Operator to slice an tensor.
blace::ops::IndexOp::IndexOp | ( | blace::ops::OpP | input, |
ml_core::BlaceIndexVec | indices, | ||
std::optional< ml_core::ColorFormatEnum > | color_format = std::nullopt |
||
) |
Constructs a slicing operator.
input | The input node. |
indices | A vector of indices to use for slicing. |
color_format | The resulting color format, optional. |