pytorch实现删除张量中的指定行列

<强>前言

在pytorch中,想删除张量中的指定行列,原本以为有个函数或者直接把某一行赋值为[]就可以,结果发现没这么简单,因此用了一个曲线救国方法,希望如果有更直接的方法,请大家指出。

<强>代码/强>

本质上是利用面具删除了指定行,然后重新指向。