索引和切片操作可以帮助我们快速提取张量中的部分数据。
1. 基本索引
PyTorch 支持与 Python 和 NumPy 类似的基本索引操作,PyTorch 中的基本索引可以通过整数值来索引张量。...[k]中的每一个[]都表示张量的一个维度,从左边开始维度依次增加,而[]中的元素值代表对应维度的索引号,「此时的索引号可以为负数,相当于从后向前索引。」...以形状为 [4, 3, 28, 28] 的图片张量为例,在 PyTorch 中图片张量的格式为 [batch_size, channel, width, hight],[4, 3, 28, 28] 的图片张量表示...如 x[0,::] 表示读取第一张图片的的所有通道的像素矩阵,其中::表示在通道维度上读取所有RGB三个通道,它等价于 x[0] 的写法。通常为了简洁,将::简写成单个冒号。...]
「还有点需要注意,在 PyTorch 中切片索引中的步长不能小于0,即不能为负数。」