2021年8月4日
关于torch.flatten的笔记
摘要： torch.nn.Flatten()可以是Sequential模型的一层，torch中定义的是： def __init__(self, start_dim: int = 1, end_dim: int = -1) -> None: super(Flatten, self).__init__() se 阅读全文
posted @ 2021-08-04 21:25 zae 阅读(528) 评论(0) 推荐(0) 编辑