pytochはtensorの指定行列を削除することを実現します。


前言
pytouchでは、tensorの指定行列を削除したいですが、関数があると思っていました。あるいは直接にある行を[]にすればいいと思いましたが、こんなに簡単ではないことが分かりました。ですから、曲線救国方法を使っています。もっと直接的な方法があれば、ご指摘ください。
コード
基本的にはmarkを利用して指定行を削除して、もう一度指します。

a = torch.rand(4, 2)
print(a)

idx = 1
a = a[torch.arange(a.size(0))!=1] 
print(a)

"""
tensor([[2.7775e-01, 3.7430e-01],
    [9.0373e-01, 8.1220e-02],
    [9.8638e-01, 8.6293e-01],
    [9.8139e-04, 9.8460e-02]])
    
tensor([[2.7775e-01, 3.7430e-01],
    [9.8638e-01, 8.6293e-01],
    [9.8139e-04, 9.8460e-02]])
"""
以上のpytouchはtensorの中の指定行列を削除することを実現しました。つまり、小編纂は皆さんに全部の内容を共有しました。皆様に参考にしてもらいたいです。