benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 953 / How to batch convert sentence lengths to masks in PyTorch?

Problem

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

prompt

Problem:

How to batch convert sentence lengths to masks in PyTorch?
For example, from

lens = [3, 5, 4]
we want to get

mask = [[0, 0, 1, 1, 1],
        [1, 1, 1, 1, 1],
        [0, 1, 1, 1, 1]]
Both of which are torch.LongTensors.


A:

<code>
import numpy as np
import pandas as pd
import torch
lens = load_data()
</code>
mask = ... # 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
953
task_id
8a0725e5-5827-5304-84a0-aa884c877437
task_key
default--test--953
task_revision_id
1
config / split
default / test

Official source

initial import