抱歉標題措辭不佳。我想做的是這樣的： 矩陣 1 是原始矩陣，矩陣 2 是矩陣 1，但每個偶數列和行都被取出。矩陣 3 是矩陣 1，但只有 1 (mod 3) 列和行。矩陣 4 相同，具有 1 (mod 4) 列和行。矩陣 5 有 1 (mod 2) 列和所有行。是否有以這種方式快速且可以利用 GPU 來操縱張量的 PyTorch 函數？這有點像 MaxPool2d，但我只需要第一個值而不是最大值。如果沒有這樣的功能，有沒有辦法手動完成但速度仍然很快？
1 回答
繁星點點滴滴
TA貢獻1803條經驗 獲得超3個贊
Matrix 5 是最容易展示的，因為你只需要沿著一個維度切片。但是您可以將兩者切分以獲得其他結果。
matrix5 = matrix1[, ::2]
這種表示法每隔一列取一次，從第 0 列開始。
添加回答
舉報
0/150
提交
取消