1. 程式人生 > 程式設計 >Pytorch高階OP操作where,gather原理

Pytorch高階OP操作where,gather原理

PyTorch是一個非常有可能改變深度學習領域前景的Python庫。我嘗試使用了幾星期PyTorch,然後被它的易用性所震驚,在我使用過的各種深度學習庫中,PyTorch是最靈活、最容易掌握的。

一、where

1)torch.where(condition,x,y) # condition是條件,滿足條件就返回x,不滿足就返回y

Pytorch高階OP操作where,gather原理

2)特點,相比for迴圈的優點是:可以佈置在GPU上執行

Pytorch高階OP操作where,gather原理

二、gather

1)官方解釋:根據指定的維度和索引值來篩選值

Pytorch高階OP操作where,gather原理

2)舉例

Pytorch高階OP操作where,gather原理

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。