python - Tensorflow GPU 中是否有一种方法可以将不同的函数应用于张量的不同列？

Question

我有一个 Tensorflow 张量，例如：

import tensorflow as tf
A = tf.constant([[3,4,5],[7,8,9])

这个张量有 3 列。我想将函数（例如平方根）应用于一列（例如第二列），而对其他列不执行任何操作。有没有办法在张量流中做到这一点？

我希望用 GPU 上的一般张量来做到这一点。张量将具有已知数量的列。将有一个长度等于列数的列表。列表的每个元素都将包含要应用于张量中相应列的函数的名称。如果不应用任何函数，则列表元素将包含 None python 对象。如果要应用函数，列表元素将包含函数的名称作为字符串。我想在 GPU 上以优化的方式做到这一点。

标签： pythontensorflow