Implemented upper and lower triangle methods in NumRuby

lib/nmatrix/numruby.rb

@@ -62,4 +62,27 @@ def self.hstack(objs)
     result
   end
 
+  def self.tri(n, m=None, k=0, nm_dtype=float)
+    if m is nil
+      m = n
+
+      m = greater_equal.outer(arange(n, nm_dtype=nm_int(0, n)),
+                              arange(-k, m-k, nm_dtype=nm_int(-k, m - k)))
+
+      # Avoid making a copy if the requested type is already bool
+      m = m.stype(nm_dtype, copy=False)
+      return m
+    end
+  end
+
+  def tril(m, k=0)
+    m = Array(m)
+    mask = tri(*m.shape[-2:], k=k, nm_dtype=nm_bool)
+  end
+
+  def triu(m, k=0)
+    m = Array(m)
+    mask = tri(*m.shape[-2:], k=k-1, nm_dtype=nm_bool)
+  end
+
 end