如何计算张量流中的阶乘？

Question

我是tensorflow的新手，我试图找到一个计算n！的函数。我看到可以使用theano函数使用theamma函数，但不适用于tensorflow。

factorial = theano.tensor.gamma(v)

我正在使用for循环将数字从n乘以1，但是我认为有一种更简便快捷的方法。我看到了与伽马分布有关的函数，但无法弄清楚如何计算阶乘。如果能指出我一些文档，将不胜感激。

这是我现在做的方式

import tensorflow as tf

factorial = tf.Variable(1, "factorial")
recursion = tf.Variable(1, "recursion")

# calculate factorial
mult = tf.multiply(recursion, factorial)
assign_fact = tf.assign(factorial, mult)

init = tf.global_variables_initializer()

with tf.Session() as sess:
    sess.run(init) 
    for i in range(2,10):
        counter = tf.assign(recursion, tf.constant(i))
        sess.run(counter)
        sess.run(assign_fact)

        print(i,"factorial is", sess.run(factorial))

    sess.close()

输出是

2 factorial is 2
3 factorial is 6
4 factorial is 24
5 factorial is 120
6 factorial is 720
7 factorial is 5040
8 factorial is 40320
9 factorial is 362880

Answer 1

试试这个：tf.exp(tf.lgamma(x + 1))。

tf.lgamma 按元素计算Gamma（x）绝对值的对数，因此指数将为您提供原始Gamma（x）值：

>>> sess.run(tf.exp(tf.lgamma(5.0)))
24.0