Pytorch阅读文档之flatten函数pytorch中flatten函数
pytorch中flatten函数torch.flatten()#展平一个连续范围的维度,输出类型为Tensortorch.flatten(input, start_dim=0, end_dim=-1) → Tensor# Parameters:input (Tensor) – 输入为Tensor#start_dim (int) – 展平的开始维度#end_dim (int) – 展...