torch.flatten¶. 火炬。 平 ( 输入,start_dim = 0,end_dim = 1 )→张量¶. 趋于平缓 输入 通过把它重塑成一维张量。如果 start_dim 或 end_dim 都通过了,只有维度 ...
確定! 回上一頁