reshape(-1, 1, 2)

-1 表示自适应该维度大小。

这里 reshape 的第一个参数为-1, 表明这个参数是根据后面的维度1X2的计算出来的。

比如如果一个 array有 NX1X2个元素,那么它的 reshape(-1, 1, 2) 方法将返回一个 shape = (n, 1, 2) 的 ndarray.

posted @ 2022-04-19 21:29  变秃了也就变强了  阅读(30)  评论(0编辑  收藏  举报  来源