Question

我是tensorflow的新手，我正在尝试找到一个计算n！的函数。我看到可以使用伽玛函数，这可能与theano一起使用，但不适用于张量流。

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

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

1 个答案: