PyTorch에서 모든 2D 서브 텐서가 대각선 행렬 인 3D 텐서를 만드는 방법은 무엇입니까?

Question

2D Tensor, index_in_batch * diag_ele을 고려하십시오. 3D 텐서 index_in_batch * Matrix을 얻으려면 어떻게해야합니까? (누가 대각선 행렬입니까, drag_ele로 구성합니까?)

torch.diag()은 입력이 1D 인 경우에만 대각 매트릭스를 구성하고, 입력이 2D 일 때 대각 요소를 반환합니다.

Answer 1

import torch 

a = torch.rand(2, 3) 
print(a) 
b = torch.eye(a.size(1)) 
c = a.unsqueeze(2).expand(*a.size(), a.size(1)) 
d = c * b 
print(d) 

출력

0.5938 0.5769 0.0555 
0.9629 0.5343 0.2576 
[torch.FloatTensor of size 2x3] 


(0 ,.,.) = 
    0.5938 0.0000 0.0000 
    0.0000 0.5769 0.0000 
    0.0000 0.0000 0.0555 

(1 ,.,.) = 
    0.9629 0.0000 0.0000 
    0.0000 0.5343 0.0000 
    0.0000 0.0000 0.2576 
[torch.FloatTensor of size 2x3x3] 

Answer 2

후방 변수에 배치하여 자동차 용 용액.

import torch 

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

b = Variable(torch.eye(a.size(1))) 
c = a.unsqueeze(2).expand(*a.size(), degree_inv.size(1)) 
b_expand = b.unsqueeze(0).expand(c.size(0), *b.size()) 
d = torch.mul(c.double(), b_expand.double()) 

print(d)