​torch.nn.Flatten(start_dim=1,end_dim=-1)作用：将连续的维度范围展平为张量。经常在nn.Sequential()中出现，一般写在某个神经网络模型之后，用于对神经网络模型...

浏览 69 次 标签： nn.Flatten()函数详解及示例