// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. #include "paddle/phi/kernels/one_hot_kernel.h" #include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/utils/data_type.h" #include "paddle/phi/kernels/funcs/math_function.h" namespace phi { template struct OneHotV2OpFunctor { const DenseTensor* in_; DenseTensor* out_; int depth_; const DeviceContext& ctx_; bool allow_out_of_range_; OneHotV2OpFunctor(const DenseTensor* in, DenseTensor* out, int depth, const DeviceContext& ctx, bool allow_out_of_range = false) : in_(in), out_(out), depth_(depth), ctx_(ctx), allow_out_of_range_(allow_out_of_range) {} template void apply() const { auto* p_in_data = in_->data(); auto numel = in_->numel(); auto* p_out_data = ctx_.template Alloc(out_); funcs::set_constant(ctx_, out_, 0.0); if (allow_out_of_range_) { for (int i = 0; i < numel; ++i) { if (p_in_data[i] >= 0 && p_in_data[i] < depth_) { *(p_out_data + i * depth_ + p_in_data[i]) = 1.0; } } } else { for (int i = 0; i < numel; ++i) { PADDLE_ENFORCE_GE( p_in_data[i], 0, phi::errors::InvalidArgument( "Illegal index value, Input(input) value should be at least 0, " "but received input (%d) less than 0", p_in_data[i])); PADDLE_ENFORCE_LT( p_in_data[i], depth_, phi::errors::InvalidArgument( "Illegal index value, Input(input) value should be less than " "Input(depth), " "but received input (%d) not less than depth (%d)", p_in_data[i], depth_)); *(p_out_data + i * depth_ + p_in_data[i]) = 1.0; } } } }; template void OneHotRawKernel(const Context& dev_ctx, const DenseTensor& x, int32_t depth, DataType dtype, bool allow_out_of_range, DenseTensor* out) { auto out_dims = out->dims(); if (out_dims[out_dims.size() - 1] == -1) { out_dims[out_dims.size() - 1] = depth; out->Resize(out_dims); } phi::VisitDataType(dtype, OneHotV2OpFunctor( &x, out, depth, dev_ctx, allow_out_of_range)); } } // namespace phi PD_REGISTER_KERNEL( one_hot_raw, CPU, ALL_LAYOUT, phi::OneHotRawKernel, int, int64_t) {}