提交 c7eaad4a 编写于 作者: F Flowingsun007

remove_onehot_consistent_tensor_limit

上级 55d32c33
......@@ -1598,10 +1598,6 @@ class OneHotFunctor {
OF_RUNTIME_ERROR() << "one_hot is only applicable to index tensor.";
}
MutableAttrMap attrs;
if (input->is_consistent()) {
OF_RUNTIME_ERROR() << "A consistent tensor can not be applied to onehot, and use "
"tensor.to_local() to convert it to local tensor first.";
}
if (num_classes == -1) {
std::vector<int32_t> axis(input->ndim());
std::iota(axis.begin(), axis.end(), 0);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册