start_dim-第一个昏暗的flatten(默认= 1)。 end_dim- last dim to flatten(默认= -1)。 例子:: >>>输入=火炬.randn(32,1,5,5)>>>米=神经网络.
確定! 回上一頁