m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix warning

Browse Source
upload-readme
chengduoZH 7 years ago
parent 0d7519171e
commit 6d0370d27a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File

8
paddle/fluid/operators/cross_entropy_op.h
Unescape View File

@ -86,10 +86,10 @@ class XeGradFunctor {
auto x_is_true_offset = sample_id * num_classes_ + label_[sample_id]; auto x_is_true_offset = sample_id * num_classes_ + label_[sample_id];
for (size_t x_offset = sample_id * num_classes_; for (size_t x_offset = sample_id * num_classes_;
x_offset < (sample_id + 1) * num_classes_; ++x_offset) { x_offset < (sample_id + 1) * num_classes_; ++x_offset) {
dx_[x_offset] = dx_[x_offset] = (x_offset != x_is_true_offset ||
(x_offset != x_is_true_offset || label_[sample_id] == ignore_index_) label_[sample_id] == static_cast<int64_t>(ignore_index_))
? static_cast<T>(0) ? static_cast<T>(0)
: -dy_[sample_id] / x_[x_offset]; : -dy_[sample_id] / x_[x_offset];
} }
} }

Write Preview
Loading…
Cancel
Save