pytorch tensor操作问题

Question

在pytorch中，我想保留一个二维tensor中每行中最大的n个值，其他位置置为0，代码该怎么写比较快？（别用for循环去写，慢死了）python不熟，谢谢各位！

Answer 1

你好，我是有问必答小助手，非常抱歉，本次您提出的有问必答问题，技术专家团超时未为您做出解答


本次提问扣除的有问必答次数，将会以问答VIP体验卡（1次有问必答机会、商城购买实体图书享受95折优惠）的形式为您补发到账户。


因为有问必答VIP体验卡有效期仅有1天，您在需要使用的时候【私信】联系我，我会为您补发。