pytorch的reshape函数中-1表示的意义

定义3X4的张量,reshape(-1, 1),你把它想象成要转换成nX1的矩阵,那是多少?没错,最后转换成了12*1的矩阵

pytorch的reshape函数中-1表示的意义_第1张图片

如果是reshape(2, -1)呢?没错,那就是2*6的矩阵了

pytorch的reshape函数中-1表示的意义_第2张图片

总结

-1代表tensor的总长度/已知的参数

你可能感兴趣的:(python,pytorch,深度学习,python)