Pytorch:将张量的前10%设置为零

Question

我有Pytorch 2d张量，它具有正态分布。是否有一种快速的方法使用Python来取消这个张量的10%的最大值？

我认为这里有两种可能的方法：

使用一些本机it

• Non-vectorized运算符(for-if)it

• Non-vectorized对

1. 平坦的张量到1d进行排序。

但这些看起来都不够快。

那么，将张量的X最大值设置为零的最快方法是什么？

Answer 1

好吧，Pytorch似乎有一个有用的运算符torch.quantile()，它在这里非常有用。

解(对于一维张量)：

import torch

x = torch.randn(100)
y = torch.tensor(0.) #new value to assign 
split_val = torch.quantile(x, 0.9)
x = torch.where(x < split_val, x, y)