pytochはtensorの指定行列を削除することを実現します。
前言
pytouchでは、tensorの指定行列を削除したいですが、関数があると思っていました。あるいは直接にある行を[]にすればいいと思いましたが、こんなに簡単ではないことが分かりました。ですから、曲線救国方法を使っています。もっと直接的な方法があれば、ご指摘ください。
コード
基本的にはmarkを利用して指定行を削除して、もう一度指します。
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の中の指定行列を削除することを実現しました。つまり、小編纂は皆さんに全部の内容を共有しました。皆様に参考にしてもらいたいです。