2.pytorch中升维度的操作及需要注意的点

二、unsqueeze、unsqueeze_和None的使用及注意in_place操作

1.unsqueeze

unsqueeze没有in_place,就不可以直接对原始张量进行赋值,需要重新对张量进行赋值。

image = torch.randn(3, 13, 13)
# 升维,变成NCHW
image = image.unsqueeze(dim=0)

2.unsqueeze_

unsqueeze_in_place,可以直接对原始张量进行赋值。

image = torch.randn(3, 13, 13)
# 升维,变成NCHW
image.unsqueeze_(dim=0) # 这里会直接改变张量的值

3.None的使用

升维可以使用None...来进行升维操作,同时需要重新对张量赋值。

image =torch.randn(3, 13, 13)
# 升维,变成NCHW
image = image[None, ...]

4. 注意,in_place是怎么一回事

  1. unsqueeze_unsqueeze实现一样的功能, 区别在于unsqueeze_in_place操作。
    unsqueeze不会对使用unsqueeze的tensor进行改变,想要获取unsqueeze后的值必须赋予个新值;unsqueeze_则会对自己改变
  2. in_place会直接对原始的张量进行改变,不用对原始张量值再次进行赋值操作,很多函数都有这个操作
  3. 【注意】:pytorch中很多的函数都是这样类似的 原理,两个函数名称相同,不同点就在于其中一个函数名后面有_,表示可以不用重新赋值,直接对改变张量的值。
上一篇:(十五)常用排序算法总结和对比


下一篇:Java程序小设计