分类：torch.flatten()函数

2022
09-03

Python torch.flatten()函数案例详解

先看函数参数：torch.flatten(input,start_dim=0,end_dim=-1)input:一个tensor，即要被“推平”的tensor。start_dim:“推平”的起始维度。end_dim: “推平”的结束维度。首先如果按照start_dim和end_dim的默认值，那么这个函数会把input推平成一个shape为[n][n]的tensor，其中nn即input中元素个数。如果我们要自己设定起始维度和结束维度呢？我们要先来看一下tensor中的shape是怎么样的：t=torch.t... 继续阅读 >