部分，类和tf.scan

Question

我可以使用functools.partial创建一个one_step函数。但是，如果partial的参数之一是一个类的实例，这会损害性能吗？

实际的elems数组很大，我在GPU上运行程序。

import numpy as np
import tensorflow as tf
from functools import partial


class B:
    def __init__(self, b):
        self.b = b


def proto_one_step(b, a, x):
    return b.b + a + x


elems = np.array([1, 2, 3, 4, 5, 6])
sum2 = tf.scan(partial(proto_one_step, B(1)), elems)


with tf.Session() as sess:
    print(sess.run(sum2))

谢谢。