我现在正在学习如何在PyTorch中构建神经网络。下面是从.py
文件中剪切出来的代码:
x = torch.unsqueeze(torch.linspace(-1, 1, 1000), dim=1)
y = x.pow(2) + 0.1*torch.normal(torch.zeros(*x.size()))
我对在x.size()
之前使用*
运算符的效用感到困惑。我尝试删除它并绘制散点图,结果证明与未删除*
的图形相同。
我还查看了https://pytorch.org/docs/stable/tensors.html中size
的官方文档,但是我无法理解。
如果您能帮忙,我将不胜感激。