コード例 #1
0
 def run_gpu_qr(self, A_val, complete=True):
     A = theano.tensor.fmatrix("A")
     fn = theano.function([A],
                          gpu_qr(A, complete=complete),
                          mode=mode_with_gpu)
     return fn(A_val)
コード例 #2
0
ファイル: test_linalg.py プロジェクト: Thrandis/Theano
 def run_gpu_qr(self, A_val, complete=True):
     A = theano.tensor.fmatrix("A")
     fn = theano.function([A], gpu_qr(A, complete=complete),
                          mode=mode_with_gpu)
     return fn(A_val)