关于nn.Flatten()函数详解及示例的资讯_nn.Flatten()函数详解及示例相关的资讯

2024-06-26 14:31:02

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