在theano矩阵中拖曳第一列

Question

我需要将每行的第一列重新排列为theano张量矩阵。

我找到的唯一方法是raw_random.shuffle_row_elements，但它会对每一行（所有列）进行混洗。

任何人都可以给我一个提示吗？

Answer 1

使用以下

theano.tensor.basic.swapaxes(y, axis1, axis2)

这被定义为：

def swapaxes(y, axis1, axis2):
    "swap axes of inputted tensor"
    y = as_tensor_variable(y)
    ndim = y.ndim
    li = range(0, ndim)
    li[axis1], li[axis2] = li[axis2], li[axis1]
    return y.dimshuffle(li)