pytorch unsqueeze函数
时间: 2023-04-28 09:00:54 浏览: 114
pyTorch.randn()、rand()、cat()、pow()、scatter_()、.squeeze() 、.unsqueeze()
5星 · 资源好评率100%
PyTorch的unsqueeze函数是用来增加一个维度的函数。它的作用是在指定的位置上增加一个维度,从而改变张量的形状。具体来说,unsqueeze函数的输入是一个张量和一个整数,整数表示要在哪个位置上增加一个维度。输出是一个新的张量,它的形状是在指定位置上增加了一个维度的形状。例如,如果输入张量的形状是(3,4),并且要在位置上增加一个维度,那么输出张量的形状就是(1,3,4)。
阅读全文