Skip to content

Commit

Permalink
added tril_operation
Browse files Browse the repository at this point in the history
  • Loading branch information
Karame committed May 4, 2021
1 parent e799f77 commit 0195add
Show file tree
Hide file tree
Showing 2 changed files with 543 additions and 10 deletions.
11 changes: 1 addition & 10 deletions phylanx/plugins/matrixops/triu_operation.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -22,16 +22,7 @@
#include <vector>

namespace phylanx { namespace execution_tree { namespace primitives {
/// \brief Return an N x M matrix with ones on the k-th diagonal and
/// zeros elsewhere.
/// \param N Number of rows in the output.
/// \param M Optional. Number of columns in the output. If None, defaults
/// to N.
/// \param k Optional. Index of the diagonal: 0 (the default) refers to the
/// main diagonal, a positive value refers to an upper diagonal,
/// and a negative value to a lower diagonal.
/// \param dtype Optional. The data-type of the returned array (default:
/// 'float')

class triu_operation
: public primitive_component_base
, public std::enable_shared_from_this<triu_operation>
Expand Down
Loading

0 comments on commit 0195add

Please sign in to comment.