未验证 提交 b1107cf4 编写于 作者: D danleifeng 提交者: GitHub

fix example bug of sampled_softmax_with_cross_entropy_cn;test=develop (#1891)

上级 ce03dbf6
...@@ -37,7 +37,7 @@ sampled_softmax_with_cross_entropy ...@@ -37,7 +37,7 @@ sampled_softmax_with_cross_entropy
import paddle.fluid as fluid import paddle.fluid as fluid
input = fluid.layers.data(name='data', shape=[256], dtype='float32') input = fluid.layers.data(name='data', shape=[256], dtype='float32')
label = fluid.layers.data(name='label', shape=[5], dtype='int64') label = fluid.layers.data(name='label', shape=[1], dtype='int64')
fc = fluid.layers.fc(input=input, size=100) fc = fluid.layers.fc(input=input, size=100)
out = fluid.layers.sampled_softmax_with_cross_entropy( out = fluid.layers.sampled_softmax_with_cross_entropy(
logits=fc, label=label, num_samples=25) logits=fc, label=label, num_samples=25)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册