view()函数

转载自:(18条消息) 【深度学习实战】从零开始深度学习(二):多层全连接神经网络与MNIST手写数字分类_喵木木的博客-CSDN博客_从零开始全连接神经

view()函数的功能与reshape类似,用来转换size大小。view()函数作用是将一个多行的Tensor,拼接成一行。

import torch
 
a = torch.Tensor(2,3)
print(a)
# tensor([[0.0000, 0.0000, 0.0000],
#        [0.0000, 0.0000, 0.0000]])
 
print(a.view(1,-1))
# tensor([[0.0000, 0.0000, 0.0000, 0.0000, 0.0000, 0.0000]])

 

posted @ 2022-11-13 12:42  实数集  阅读(101)  评论(0编辑  收藏  举报