eric-haibin-lin closed pull request #12701: Add FListInputNames attribute to
softmax_cross_entropy
URL: https://github.com/apache/incubator-mxnet/pull/12701
This is a PR merged from a forked repository.
As GitHub hides the original diff on merge, it is displayed below for
the sake of provenance:
As this is a foreign pull request (from a fork), the diff is supplied
below (as it won't show otherwise due to GitHub magic):
diff --git a/src/operator/loss_binary_op.cc b/src/operator/loss_binary_op.cc
index c1fedb3de61..df8576cfbb8 100644
--- a/src/operator/loss_binary_op.cc
+++ b/src/operator/loss_binary_op.cc
@@ -67,6 +67,10 @@ Example::
})
.set_attr<FCompute>("FCompute<cpu>", SoftmaxCrossEntropyForward<cpu>)
.set_attr<nnvm::FGradient>("FGradient",
ElemwiseGradUseIn{"_backward_softmax_cross_entropy"})
+.set_attr<nnvm::FListInputNames>("FListInputNames",
+ [](const NodeAttrs& attrs) {
+ return std::vector<std::string>{"data", "label"};
+})
.add_argument("data", "NDArray-or-Symbol", "Input data")
.add_argument("label", "NDArray-or-Symbol", "Input label");
diff --git a/tests/python/unittest/test_operator.py
b/tests/python/unittest/test_operator.py
index b5a7303195f..b17562c1d94 100644
--- a/tests/python/unittest/test_operator.py
+++ b/tests/python/unittest/test_operator.py
@@ -6970,6 +6970,25 @@ def test_invalid_depth_dim():
test_invalid_block_size()
test_invalid_depth_dim()
+
+@with_seed()
+def test_softmax_cross_entropy():
+ def f_sm_ce(data, label):
+ return np.sum(-np.log(data) * label)
+
+ data = mx.sym.Variable('data')
+ label = mx.sym.Variable('label')
+ sym = mx.sym.softmax_cross_entropy(data=data, label=label)
+ num_labels = random.randint(100, 200)
+ batch_size = random.randint(100, 200)
+ np_data = rand_ndarray((batch_size, num_labels), stype='default').asnumpy()
+ np_sm = np_softmax(np_data)
+ np_label = np.random.randint(0, num_labels, (batch_size, ))
+ np_one_hot_label = np.zeros((batch_size, num_labels))
+ np_one_hot_label[np.arange(batch_size), np_label] = 1.
+ check_symbolic_forward(sym, {'data' : np_data, 'label' : np_label},
[np.array([f_sm_ce(np_sm, np_one_hot_label)])], rtol=1e-3, atol=1e-5)
+
+
@with_seed()
def test_invalid_kernel_size():
invalid_kernel_size = 28
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
