pytorch tensor 扩充一个维度,避免使用unsqueeze()

pytorch 将tensor膨胀出一个额外的维度(unsqueeze()的功能)

在需要膨胀(本来不存在的维度上)用None取值

 # Ps size: 10
 a = Ps[:, None]
# a size: 10x1

# 等价于
b = Ps.unsqueeze(-1)
# b size: 10x1

避免使用Ps.unsqueeze(),尺度扩充后可以直接进行计算,无需多一行操作。(算是个trick?

你可能感兴趣的:(pytorch,pytorch,人工智能,python)