从Pytorch中的4D张量求零指数
如果你正在与Pytorch合作，我有一个有趣的问题要问你 我有一批形状为3、224、224的图像。如果批量大小是64，那么最后的张量的形状是64，3224，224 现在，问题来了。假设此批处理中的某些图像仅填充了零。找出哪些批次索引只有零的最快方法是什么 我不想为此创建for循环，因为它很慢从Pytorch中的4D张量求零指数,pytorch,Pytorch,如果你正在与Pytorch合作，我有一个有趣的问题要问你 我有一批形状为3、224、224的图像。如果批量大小是64，那么最后的张量的形状是64，3224，224 现在，问题来了。假设此批处理中的某些图像仅填充了零。找出哪些批次索引只有零的最快方法是什么 我不想为此创建for循环，因为它很慢 谢谢你的回答。一种更便宜的方法是假设只有空图像的sum=0，我认为这是非常合理的 进口火炬 t=火炬。rand64,3224224 t[10]=0 s=t.view64，-1.sumdim=-1 零指数=s
谢谢你的回答。一种更便宜的方法是假设只有空图像的sum=0，我认为这是非常合理的 进口火炬 t=火炬。rand64,3224224 t[10]=0 s=t.view64，-1.sumdim=-1 零指数=s==0.nonzero.item 10