benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 968 / I have the following torch tensor:

Problem

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

prompt

Problem:

I have the following torch tensor:

tensor([[-0.2,  0.3],
    [-0.5,  0.1],
    [-0.4,  0.2]])
and the following numpy array: (I can convert it to something else if necessary)

[1 0 1]
I want to get the following tensor:

tensor([0.3, -0.5, 0.2])
i.e. I want the numpy array to index each sub-element of my tensor. Preferably without using a loop.

Thanks in advance


A:

<code>
import numpy as np
import pandas as pd
import torch
t, idx = load_data()
assert type(t) == torch.Tensor
assert type(idx) == np.ndarray
</code>
result = ... # put solution in this variable
BEGIN SOLUTION
<code>

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
968
task_id
14db90bc-060f-5d70-9621-8e0569de30ab
task_key
default--test--968
task_revision_id
1
config / split
default / test

Official source

initial import