torch.flatten(input, start_dim=0, end_dim=-1) flattens a contiguous range of dimensions in a tensor. torch.reshape(input, shape) returns a ...
確定! 回上一頁