如何在theano中的张量中仅设置单个元素？

Question

目前我有这个代码设置张量输出的第一个元素。

        a = T.set_subtensor(output[0][0], rotation[0][0])
        output = T.set_subtensor(output[0],a)

        # outputs as expected:
        # [
        #   [.934, 0],
        #   [0,    0]
        # ]

不幸的是：

       output = T.set_subtensor(output[0][0], rotation[0][0])
       # outputs single array [.934, 0]... this is not what I want

只输出不是我想要的第一行。是否有可能创建一个结合前两行的语句来为我提供我想要的矩阵？

Answer 1

您最有可能想要替换

output = T.set_subtensor(output[0][0], rotation[0][0])

与

output = T.set_subtensor(output[0, 0], rotation[0, 0])

如果可能的话。