Tensor的切片

Tensor的切片_第1张图片

chunk split

Tensor的切片_第2张图片

注意 chunk第二个参数表示 分成几块;split第二个参数表示 分成多大的块。

chunk分块时,每块的平均数会向上取整,如本例:4/3向上取整,每个0维度上,分2个元素,所以才会出现传参数3,要求分成3块,最后却只分成2块。

split的另外一种传参方式

Tensor的切片_第3张图片

 

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