torch 中的setNumInputDims()函数
输入:
minibatch = torch.Tensor(5, 2, 3, 5)
m = nn.View(-1):setNumInputDims(1)
m:forward(minibatch)
---------------------------------------------------------------------------------------------------...
原创
2019-05-14 10:33:07 ·
503 阅读 ·
0 评论