Pytorch(3)

1.broadcasting(自动扩展)

为了使得不同维度的tensor 进行相加

a = [4,32,14,14]
b = [14,14]=>[1,1,14,14]=>[4,32,14,14]

不能使用broadcasting

a = [4,32,14,14]
b = [2,32,14,14]
Dim 0 has dim,can NOT insert and expand to same
Dim 0 has distinct dim,NOT size 1
NOT brodcasting-able

2.合并与切割

(1)cat 1.维度相同 2. 只有一个维度的尺寸不一样 其他维度尺寸都一样 例如 [4,32,8] 和 [5,32,8] 第一列代表班级 第二列代表班级 人数 第三列代表课程数量 只有才能进行cat

Pytorch(3)_第1张图片

(2)stack创建新的维度 例如[8,63] [9,66] 表示[班级 ,每个班的数学成绩] 需要进行这样的合并[8,63,9,66] 而使用cat变成[17,129]

Pytorch(3)_第2张图片

(3)split 拆分 按长度进行拆封 a.split([2,3],dim=0) 表示将一个维度 前两个放在一起 后三个放在一起

Pytorch(3)_第3张图片

(4) chunk 通过数量进行拆分

Pytorch(3)_第4张图片

你可能感兴趣的:(学习笔记,pytorch,深度学习,人工智能)