1分钟理解pytorch的reshape函数中-1表示的意义

先说答案,reshape函数中-1代表的是n,什么意思呢,函数中另一个参数决定了-1的值,看下面三张图就很容易理解了 定义34的张量,reshape(-1, 1),你把它想象成要转换成n1的矩阵,那是多少?没错,最后转换成了12*1的矩阵 同理,如果是reshape(1, -1)呢?那就是1*12了 最后的一个更加直观,如果是reshape(2, -1)呢?没错,那就是2*6的矩阵了,是不是1分钟
相关文章
相关标签/搜索