pytorch nn.flatten()函数
0
-
pytorch中nn.Flatten()函数详解及示例
torch.nn.Flatten(start_dim=1, end_dim=- 1) 作用：将连续的维度范围展平为张量。 经常在nn.Sequential()中出现，一般写在某个神经网络模型之后，用于对神经网络模型的输出进行处理，得到tensor类型的数据[详细]
2023-01-07 09:34 分类：开发
加载中,请稍侯......