Python torch.flatten()函数案例详解. 先看函数参数:. torch.flatten(input, start_dim=0, end_dim=-1). input: 一个tensor,即要被“推平”的tensor。
確定! 回上一頁