Browse Source

add comment about nonzero op

tags/v1.6.0
zhengjun10 4 years ago
parent
commit
7d80d4d7ac
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/core/ops/non_zero.h

+ 4
- 0
mindspore/core/ops/non_zero.h View File

@@ -24,9 +24,13 @@
namespace mindspore {
namespace ops {
constexpr auto kNameNonZero = "NonZero";
/// \brief Calculate tensor not zero index, by default.
/// Refer to Python API @ref mindspore.ops.NonZero for more details.
class MS_CORE_API NonZero : public PrimitiveC {
public:
/// \brief Constructor.
NonZero() : PrimitiveC(kNameNonZero) {}
/// \brief Destructor.
~NonZero() = default;
MS_DECLARE_PARENT(NonZero, PrimitiveC);
};


Loading…
Cancel
Save