torch.view()
如果你想改变一个 tensor 的大小或者形状,你可以使用torch.view。
在pytorch中view函数的作用为重构张量的维度,相当于numpy中resize()的功能。view()返回的数据和传入的tensor一样,只是形状不同。
-1参数
-1在这里的意思是让电脑帮我们计算。
view()返回的tensor和传入的tensor共享内存。
版权声明:本文为qq_43539854原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。