blace.ai
|
#include <public_ops.h>
Public Member Functions | |
IndexPutOp (blace::ops::OpP input, ml_core::BlaceIndexVec indices, blace::ops::OpP val) | |
IndexPutOp (blace::ops::OpP input, ml_core::BlaceIndexVec indices, double val) | |
IndexPutOp (blace::ops::OpP input, ml_core::BlaceIndexVec indices, float val) | |
IndexPutOp (blace::ops::OpP input, ml_core::BlaceIndexVec indices, int val) | |
Operator to put values into a tensor.
blace::ops::IndexPutOp::IndexPutOp | ( | blace::ops::OpP | input, |
ml_core::BlaceIndexVec | indices, | ||
blace::ops::OpP | val | ||
) |
Puts a value from a blace::ops::OpP into the input tensor at the specified indices.
input | |
indices | |
val |
blace::ops::IndexPutOp::IndexPutOp | ( | blace::ops::OpP | input, |
ml_core::BlaceIndexVec | indices, | ||
double | val | ||
) |
Puts a double into the input tensor at the specified indices.
input | |
indices | |
val |
blace::ops::IndexPutOp::IndexPutOp | ( | blace::ops::OpP | input, |
ml_core::BlaceIndexVec | indices, | ||
float | val | ||
) |
Puts a float into the input tensor at the specified indices.
input | |
indices | |
val |
blace::ops::IndexPutOp::IndexPutOp | ( | blace::ops::OpP | input, |
ml_core::BlaceIndexVec | indices, | ||
int | val | ||
) |
Puts an integer into the input tensor at the specified indices.
input | |
indices | |
val |