x.view(-1,4)

1、问题描述

在构建神经网络的时候,经常会用到x.view()函数,实际上view()类似于reshape()的用法,将张量重新规划格式,本文将简单介绍这个函数的用法。

2、代码实例

import torch

a = torch.arange(1,17)
print(a)

a = a.view(-1,4)
print(a)

这里view的第一个参数有时会是-1,-1代表不确定,行数将由张量的长度除以列数决定,也就是说

a.view(-1,4) == a.view(4,4)
a.view(-1,8) == a.view(2,8)

猜你喜欢

转载自blog.csdn.net/xdg15294969271/article/details/121675101