benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 971 / I have the tensors:

Problem

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

prompt

Problem:

I have the tensors:

ids: shape (70,1) containing indices like [[1],[0],[2],...]

x: shape(70,3,2)

ids tensor encodes the index of bold marked dimension of x which should be selected. I want to gather the selected slices in a resulting vector:

result: shape (70,2)

Background:

I have some scores (shape = (70,3)) for each of the 3 elements and want only to select the one with the highest score. Therefore, I used the function

ids = torch.argmax(scores,1,True)
giving me the maximum ids. I already tried to do it with gather function:

result = x.gather(1,ids)
but that didn't work.


A:

<code>
import numpy as np
import pandas as pd
import torch
ids, x = load_data()
</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
971
task_id
34162342-f548-5747-99a9-1fd2556a66b2
task_key
default--test--971
task_revision_id
1
config / split
default / test

Official source

initial import