Python Gemvの例

コード例 #1

ファイル: blas.py プロジェクト: caglar/Theano

 def make_node(self, y, alpha, A, x, beta):
     Gemv.make_node(self, y, alpha, A, x, beta)
     A = as_gpuarray_variable(A)
     x = as_gpuarray_variable(x)
     y = as_gpuarray_variable(y)
     assert A.dtype == x.dtype == y.dtype
     return Apply(self, [y, alpha, A, x, beta], [y.type()])

コード例 #2

ファイル: blas.py プロジェクト: MatthieuCourbariaux/Theano

 def make_node(self, y, alpha, A, x, beta):
     res = Gemv.make_node(self, y, alpha, A, x, beta)
     A = as_gpuarray_variable(A)
     x = as_gpuarray_variable(x)
     y = as_gpuarray_variable(y)
     assert A.dtype == x.dtype == y.dtype
     return Apply(self, [y, alpha, A, x, beta], [y.type()])

コード例 #3

ファイル: blas.py プロジェクト: SamuelZeng/Theano

 def make_node(self, y, alpha, A, x, beta):
     res = Gemv.make_node(self, y, alpha, A, x, beta)
     A = as_gpuarray_variable(A)
     x = as_gpuarray_variable(x)
     y = as_gpuarray_variable(y)
     return Apply(self, [y, alpha, A, x, beta], [y.type()])