pytorch 中flatten函数torch.flatten()#展平一个连续范围的维度,输出类型为Tensortorch.flatten(input, start_dim=0, end_dim=-1) → Tensor# ...
確定! 回上一頁