- Python 深度学习
- 吕云翔 刘卓然 关捷雄等编著
- 98字
- 2025-02-28 02:11:34
4.2 Tensor的索引和切片
Tensor不仅支持基本的索引和切片操作,还支持ndarray中的高级索引(整数索引和布尔索引)操作。


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

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