benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 956 / Consider I have 2D Tensor, index_in_batch * diag_ele. How can I get a 3D Tensor…

Problem

Answer published by the source. Consult the official source to check your work against its answer.

prompt

Problem:

Consider I have 2D Tensor, index_in_batch * diag_ele. How can I get a 3D Tensor index_in_batch * Matrix (who is a diagonal matrix, construct by drag_ele)?

The torch.diag() construct diagonal matrix only when input is 1D, and return diagonal element when input is 2D.


A:

<code>
import numpy as np
import pandas as pd
import torch
Tensor_2D = load_data()
def Convert(t):
    # return the solution in this function
    # result = Convert(t)
    ### BEGIN SOLUTION

library

Pytorch

Discussion

Discussion

No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

See answer Answer published by the source

Artifacts

Code, notes and reproducible work shared by participants. Files are served from a separate origin.

No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

Source and history

upstream_id / Original problem ID
956
task_id
93fec6b2-e543-5f53-a444-23b9b00acf4b
task_key
default--test--956
task_revision_id
1
config / split
default / test

Official source

initial import