|
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. |