This commit is contained in:
v-zhimao 2019-05-20 00:17:54 -07:00
Родитель 76424d9dbc
Коммит f28fd97a8b
1 изменённых файлов: 5 добавлений и 5 удалений

Просмотреть файл

@ -5481,19 +5481,19 @@ void CPUMatrix<ElemType>::GetDenseLabelsFromOneHot(const CPUMatrix<ElemType>& on
for (long i = 0; i < (num & ~3); i += 4)
{
if (oneHotLabelsPtr[i] > (ElemType)0.5)
labels[i / rows] = i % rows;
labelsPtr[i / rows] = i % rows;
if (oneHotLabelsPtr[i + 1] > (ElemType)0.5)
labels[(i + 1) / rows] = (i + 1) % rows;
labelsPtr[(i + 1) / rows] = (i + 1) % rows;
if (oneHotLabelsPtr[i + 2] > (ElemType)0.5)
labels[(i + 2) / rows] = (i + 2) % rows;
labelsPtr[(i + 2) / rows] = (i + 2) % rows;
if (oneHotLabelsPtr[i + 3] > (ElemType)0.5)
labels[(i + 3) / rows] = (i + 3) % rows;
labelsPtr[(i + 3) / rows] = (i + 3) % rows;
}
// handle remaining stuffs
for (long i = num & ~3; i < num; i++)
{
if (oneHotLabelsPtr[i] > (ElemType)0.5)
labels[i / rows] = i % rows;
labelsPtr[i / rows] = i % rows;
}
}