转载自:(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]])
标签:tensor,torch,print,0.0000,view,函数 From: https://www.cnblogs.com/ycliu/p/16885794.html