zero diagonal of a pytorch tensor?

前端 未结 0 934
Happy的楠姐
Happy的楠姐 2021-01-14 09:31

Is there a simple way to zero a diagonal of a pytorch tensor?

for example I have:

tensor([[2.7183, 0.4005, 2.7183, 0.5236],
    [0.4005, 2.7183, 0.4004         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题