2D Tensor, index_in_batch * diag_ele
을 고려하십시오. 3D 텐서 index_in_batch * Matrix
을 얻으려면 어떻게해야합니까? (누가 대각선 행렬입니까, drag_ele로 구성합니까?)PyTorch에서 모든 2D 서브 텐서가 대각선 행렬 인 3D 텐서를 만드는 방법은 무엇입니까?
torch.diag()
은 입력이 1D 인 경우에만 대각 매트릭스를 구성하고, 입력이 2D 일 때 대각 요소를 반환합니다.
정말 고마워요, 제가 정확히 원하는 것입니다. –
그런데 변수에 적용 할 때 자동으로 역으로 계산할 수있는 또 다른 단계가 있습니까? –
@ 刘钦 卿 나는 당신의 질문을 이해하지 못합니다. 새로운 질문이있는 경우 별도의 질문으로 SO에 게시하십시오 (세부 정보 포함). 또한, 나의 대답이 당신을 도왔다면, 당신은 녹색 표시를 지워서 그것을 받아 들일 수 있습니다, 감사합니다. –