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