We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0d75191 commit 6d0370dCopy full SHA for 6d0370d
paddle/fluid/operators/cross_entropy_op.h
@@ -86,10 +86,10 @@ class XeGradFunctor {
86
auto x_is_true_offset = sample_id * num_classes_ + label_[sample_id];
87
for (size_t x_offset = sample_id * num_classes_;
88
x_offset < (sample_id + 1) * num_classes_; ++x_offset) {
89
- dx_[x_offset] =
90
- (x_offset != x_is_true_offset || label_[sample_id] == ignore_index_)
91
- ? static_cast<T>(0)
92
- : -dy_[sample_id] / x_[x_offset];
+ dx_[x_offset] = (x_offset != x_is_true_offset ||
+ label_[sample_id] == static_cast<int64_t>(ignore_index_))
+ ? static_cast<T>(0)
+ : -dy_[sample_id] / x_[x_offset];
93
}
94
95
0 commit comments