4.2 Tensor的索引和切片

Tensor不仅支持基本的索引和切片操作,还支持ndarray中的高级索引(整数索引和布尔索引)操作。

torch.where(condition, x, y)判断condition的条件是否满足,当某个元素满足时,则返回对应矩阵x相同位置的元素,否则返回矩阵y的元素。