cythonの型定義

Question

cythonファイルに型を定義したいです
もともとのpyxファイルと
自分で型を定義したときのコードをのせます

gamma1.pyx

from math import exp
def gamma(word_all, word_topic):
    c = exp(w_all+w_topic) / (sum([exp(wall + wtopic) for wall, wtopic in zip(word_all, word_topic)]))
    return c
-------------------------------------------------------
from math import exp

cdef gamma(float word_all, float word_topic):
    c = exp(w_all+w_topic) / (sum([exp(wall + wtopic) for wall, wtopic in zip(word_all, word_topic)]))
    return c

--------------------------------------------------------
引数はこのような形です
In [2]: word_all
Out[2]: [0.2, 0.2, 0.2, 0.2, 0.2]
In [7]: word_topic
Out[7]:
[[0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02],
 [0.02, 0.02, 0.02, 0.02, 0.02]]

このような型の定義の仕方で大丈夫でしょうか

Comments

[shimizukawa]

https://teratail.com/questions/84214 で解答した定義の仕方を参照してください。

[kohekoh]

いまいちわかっていないのですが、配列はdtypeで定義できるのですか？

[kohekoh]

一個前のは引数がarrayだったのでdtypeが使えたんですかね？、今回同じようにやると、ビルドはできるのですが、その関数を用いたコードを実行した際にエラーがでます

Answer 1

これでできました

from math import exp
cimport numpy as np
ctypedef np.float64_t DTYPE_t

def gamma(np.ndarray[DTYPE_t, ndim=1] word_all, np.ndarray[DTYPE_t, ndim=1] word_topic, float w_all, float w_topic):
    cdef DTYPE_t c = exp(w_all+w_topic) / (sum([exp(wall + wtopic) for wall, wtopic in zip(word_all, word_topic)]))
    return c