Python 我如何在theano中只设置张量中的一个元素？

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

        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

---

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

您很可能想要替换它

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

与

如果可能的话

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