end_dim: “推平”的结束维度。 首先如果按照start_dim 和end_dim 的默认值,那么这个函数会把input 推平成一个shape 为[n][n] 的tensor, ...
確定! 回上一頁