pymc.math.flatten

pymc.math.flatten(x, ndim=1)

返回数组的一个副本，将其折叠为一维。

通过保持 x 的前 outdim-1 维的大小不变，并将 x 的最后一维的大小设置为其剩余维大小的乘积，来重塑变量 x。

参数:

xpytensor.tensor.var.TensorVariable

要重塑的变量。

ndimint

返回变量的维度数量。默认值是 1。

返回:

pytensor.tensor.var.TensorVariable

具有 outdim 维度的展平变量