unpooling.cc 3.7 KB
Newer Older
S
sweetsky0901 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

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

S
sweetsky0901 已提交
15
#include "paddle/operators/math/unpooling.h"
S
sweetsky0901 已提交
16 17 18 19 20 21 22

namespace paddle {
namespace operators {
namespace math {

// All tensors are in NCHW format
template <typename T>
S
sweetsky0901 已提交
23
class Unpool2d_MaxFunctor<platform::CPUPlace, T> {
S
sweetsky0901 已提交
24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
 public:
  void operator()(const platform::DeviceContext& context,
                  const framework::Tensor& input,
                  const framework::Tensor& indices,
                  framework::Tensor * output) {
    const int batch_size = input.dims()[0];
    const int input_height = input.dims()[2];
    const int input_width = input.dims()[3];
    const int output_channels = output->dims()[1];
    const int output_height = output->dims()[2];
    const int output_width = output->dims()[3];

    int input_feasize = input_height * input_width;
    int output_feasize = output_height * output_width;
    const T* input_data = input.data<T>();
S
sweetsky0901 已提交
39
    const int * indices_data = indices.data<int>();
S
sweetsky0901 已提交
40 41 42 43 44 45
    T* output_data = output->mutable_data<T>(context.GetPlace());

    for (int b = 0; b < batch_size; ++b) {
      for (int c = 0; c < output_channels; ++c) {
        for (int i = 0; i < input_feasize; ++i) {
          int index =  indices_data[i];
S
sweetsky0901 已提交
46
          // PADDLE_ENFORCE(index < output_feasize, "err index in unpooling!");
S
sweetsky0901 已提交
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
          output_data[index] = input_data[i];
        }
        input_data += input_feasize;
        indices_data += input_feasize;
        output_data += output_feasize;
      }
    }
  }
};



template <class T>
class Unpool2d_MaxGradFunctor<platform::CPUPlace, T> {
public:
  void operator()(const platform::DeviceContext& context,
                  const framework::Tensor& input,
                  const framework::Tensor& indices,
                  framework::Tensor * input_grad,
                  const framework::Tensor& output,
                  const framework::Tensor& output_grad) {
    const int batch_size = input.dims()[0];
    const int input_height = input.dims()[2];
    const int input_width = input.dims()[3];
S
sweetsky0901 已提交
71 72 73
    const int output_channels = output.dims()[1];
    const int output_height = output.dims()[2];
    const int output_width = output.dims()[3];
S
sweetsky0901 已提交
74 75 76

    int input_feasize = input_height * input_width;
    int output_feasize = output_height * output_width;
S
sweetsky0901 已提交
77
    const int* indices_data = indices.data<int>();
S
sweetsky0901 已提交
78 79 80 81 82 83
    const T* output_grad_data = output_grad.data<T>();

    T* input_grad_data = input_grad->mutable_data<T>(context.GetPlace());

    for (int b = 0; b < batch_size; ++b) {
      for (int c = 0; c < output_channels; ++c) {
S
sweetsky0901 已提交
84
        for (int i = 0; i < input_feasize; ++i) {
S
sweetsky0901 已提交
85
          int index = indices_data[i];
S
sweetsky0901 已提交
86
          // PADDLE_ENFORCE(index < output_feasize, "err index in unpooling!");
S
sweetsky0901 已提交
87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
          input_grad_data[i] = output_grad_data[index];
        }
        input_grad_data += input_feasize;
        indices_data += input_feasize;
        output_grad_data += output_feasize;
      }
    }
  }
};

template class Unpool2d_MaxGradFunctor<platform::CPUPlace, float>;
template class Unpool2d_MaxGradFunctor<platform::CPUPlace, double>;
template class Unpool2d_MaxFunctor<platform::CPUPlace, float>;
template class Unpool2d_MaxFunctor<platform::CPUPlace, double>;

}  // namespace math
}  // namespace operators
}  // namespace paddle