def geoseries_sum(r, t_end, t_start):
"""
Sum of r**t from t=t_start to t=t_end, inclusive
:param r:
:param t_end:
:param t_start:
:return:
"""
# return ifelse(tt.eq(r, 1), (t_end-t_start+1).astype(theano.config.floatX), (r**(t_end+1)-r**t_start)/(r-1))
return ifelse(tt.eq(r, 1), (t_end-t_start+1).astype(theano.config.floatX), (r**(t_end+1)-r**t_start)/(r-1))
评论列表
文章目录