德雷克海峡扁平

Question

为什么德雷克沿着通道维度而不是批处理维度扁平？

对于n，c向量，numpy/torch/tensorflow有一个平坦运算，它的逆运算是一个整形运算。

这个扁平操作沿批处理维度"n“扁平，而Drake沿"c”扁平。

是否有一些内置的功能，以获得沿批尺寸的扁平？

例如，我想访问：

prog.initial_guess()

但是，我希望它沿着批处理维度被压平，根据原始的矩阵形状的决策变量。

Answer 1

Drake是用C++编写的，带有python绑定。在C++中，我们使用特征作为线性代数库。矩阵默认为列大调，即按mat0，0，mat1，0，mat2，0，.，mat0，1，mat1，1，.的顺序存储数据。这与numpy/py火炬存储矩阵的方式不同，后者默认为行主行。

特别是关于获得矩阵变量的初始猜测的问题，您可以调用该函数。

initial_guess = prog.GetInitialGuess(variable_matrix)

它将返回浮点数的numpy矩阵，其中initial_guess[i, j]是变量variable_matrix[i, j]的初始猜测值，因此不需要担心矩阵的扁平化/整形。