先看函数参数： torch.flatten(input, start_dim=0, end_dim=-1) input: 一个 tensor，即要被“推平”的 tensor。 … Continue reading 关于torch.flatten的笔记
先看函数参数： torch.flatten(input, start_dim=0, end_dim=-1) input: 一个 tensor，即要被“推平”的 tensor。 … Continue reading 关于torch.flatten的笔记