
上QQ阅读APP看书,第一时间看更新
4.2 Tensor的索引和切片
Tensor不仅支持基本的索引和切片操作,还支持ndarray中的高级索引(整数索引和布尔索引)操作。


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

Tensor不仅支持基本的索引和切片操作,还支持ndarray中的高级索引(整数索引和布尔索引)操作。
torch.where(condition, x, y)判断condition的条件是否满足,当某个元素满足时,则返回对应矩阵x相同位置的元素,否则返回矩阵y的元素。