| 57 | } |
| 58 | |
| 59 | Tensor Erase(Tensor &input, Tensor &anchor, Tensor &erasing, Tensor &values, Tensor &imgIdx, bool random, |
| 60 | unsigned int seed, std::optional<Stream> pstream) |
| 61 | { |
| 62 | Tensor output = Tensor::Create(input.shape(), input.dtype()); |
| 63 | |
| 64 | return EraseInto(output, input, anchor, erasing, values, imgIdx, random, seed, pstream); |
| 65 | } |
| 66 | |
| 67 | ImageBatchVarShape EraseVarShapeInto(ImageBatchVarShape &output, ImageBatchVarShape &input, Tensor &anchor, |
| 68 | Tensor &erasing, Tensor &values, Tensor &imgIdx, bool random, unsigned int seed, |