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