2021年3月25日
关于torch.flatten的笔记
摘要： 先看函数参数： torch.flatten(input, start_dim=0, end_dim=-1) input: 一个 tensor，即要被“推平”的 tensor。 start_dim: “推平”的起始维度。 end_dim: “推平”的结束维度。 首先如果按照 start_dim 和 e 阅读全文
posted @ 2021-03-25 13:31 Moese 阅读(151) 评论(0) 推荐(0)