Pytorch-unsqueeze

unsqueeze


作用是用于增加维度,操作是针对于tensor张量,增加一个维数为1的维度。


unsqueeze的用法通过tensor.unsqueeze(dim)进行维度扩张。dim是维度,如果要扩张第一维就是tensor.unsqueeze(0)

如果要在第二维扩张就是tensor.unsqueeze(1)

最后一维就是tensor.unsqueeze(-1) 

unsqueeze_和unsqueeze实现一样的功能, 区别在于unsqueeze_是in_place操作,即unsqueeze不会对使用unsqueeze的tensor进行改变,想要获取unsqueeze后的值必须赋予个新值,unsqueeze_则会对自己改变 

squeeze
作用是用于压缩维度,操作是针对于tensor张量。与unsqueeze一样,传入参数为维度,只是作用变成了压缩(挤压)维度,去掉一个维数为1的维度.

你可能感兴趣的:(Torch)