Flatten. class torch.nn. Flatten (start_dim=1, end_dim=- 1)[source]. Flattens a contiguous range of dims into a tensor. For use with Sequential .
確定! 回上一頁