How to batch convert sentence lengths to masks in PyTorch?












2















For example, from



lens = [3, 5, 4]


we want to get



mask = [[1, 1, 1, 0, 0],
[1, 1, 1, 1, 1],
[1, 1, 1, 1, 0]]


Both of which are torch.LongTensors.










share|improve this question





























    2















    For example, from



    lens = [3, 5, 4]


    we want to get



    mask = [[1, 1, 1, 0, 0],
    [1, 1, 1, 1, 1],
    [1, 1, 1, 1, 0]]


    Both of which are torch.LongTensors.










    share|improve this question



























      2












      2








      2


      1






      For example, from



      lens = [3, 5, 4]


      we want to get



      mask = [[1, 1, 1, 0, 0],
      [1, 1, 1, 1, 1],
      [1, 1, 1, 1, 0]]


      Both of which are torch.LongTensors.










      share|improve this question
















      For example, from



      lens = [3, 5, 4]


      we want to get



      mask = [[1, 1, 1, 0, 0],
      [1, 1, 1, 1, 1],
      [1, 1, 1, 1, 0]]


      Both of which are torch.LongTensors.







      nlp pytorch






      share|improve this question















      share|improve this question













      share|improve this question




      share|improve this question








      edited Nov 20 '18 at 23:58







      0472037

















      asked Nov 20 '18 at 23:47









      04720370472037

      517




      517
























          1 Answer
          1






          active

          oldest

          votes


















          4














          One way that I found is:



          torch.arange(max_len).expand(len(lens), max_len) < lens.unsqueeze(1)


          Please share if there are better ways!






          share|improve this answer























            Your Answer






            StackExchange.ifUsing("editor", function () {
            StackExchange.using("externalEditor", function () {
            StackExchange.using("snippets", function () {
            StackExchange.snippets.init();
            });
            });
            }, "code-snippets");

            StackExchange.ready(function() {
            var channelOptions = {
            tags: "".split(" "),
            id: "1"
            };
            initTagRenderer("".split(" "), "".split(" "), channelOptions);

            StackExchange.using("externalEditor", function() {
            // Have to fire editor after snippets, if snippets enabled
            if (StackExchange.settings.snippets.snippetsEnabled) {
            StackExchange.using("snippets", function() {
            createEditor();
            });
            }
            else {
            createEditor();
            }
            });

            function createEditor() {
            StackExchange.prepareEditor({
            heartbeatType: 'answer',
            autoActivateHeartbeat: false,
            convertImagesToLinks: true,
            noModals: true,
            showLowRepImageUploadWarning: true,
            reputationToPostImages: 10,
            bindNavPrevention: true,
            postfix: "",
            imageUploader: {
            brandingHtml: "Powered by u003ca class="icon-imgur-white" href="https://imgur.com/"u003eu003c/au003e",
            contentPolicyHtml: "User contributions licensed under u003ca href="https://creativecommons.org/licenses/by-sa/3.0/"u003ecc by-sa 3.0 with attribution requiredu003c/au003e u003ca href="https://stackoverflow.com/legal/content-policy"u003e(content policy)u003c/au003e",
            allowUrls: true
            },
            onDemand: true,
            discardSelector: ".discard-answer"
            ,immediatelyShowMarkdownHelp:true
            });


            }
            });














            draft saved

            draft discarded


















            StackExchange.ready(
            function () {
            StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53403306%2fhow-to-batch-convert-sentence-lengths-to-masks-in-pytorch%23new-answer', 'question_page');
            }
            );

            Post as a guest















            Required, but never shown

























            1 Answer
            1






            active

            oldest

            votes








            1 Answer
            1






            active

            oldest

            votes









            active

            oldest

            votes






            active

            oldest

            votes









            4














            One way that I found is:



            torch.arange(max_len).expand(len(lens), max_len) < lens.unsqueeze(1)


            Please share if there are better ways!






            share|improve this answer




























              4














              One way that I found is:



              torch.arange(max_len).expand(len(lens), max_len) < lens.unsqueeze(1)


              Please share if there are better ways!






              share|improve this answer


























                4












                4








                4







                One way that I found is:



                torch.arange(max_len).expand(len(lens), max_len) < lens.unsqueeze(1)


                Please share if there are better ways!






                share|improve this answer













                One way that I found is:



                torch.arange(max_len).expand(len(lens), max_len) < lens.unsqueeze(1)


                Please share if there are better ways!







                share|improve this answer












                share|improve this answer



                share|improve this answer










                answered Nov 20 '18 at 23:58









                04720370472037

                517




                517
































                    draft saved

                    draft discarded




















































                    Thanks for contributing an answer to Stack Overflow!


                    • Please be sure to answer the question. Provide details and share your research!

                    But avoid



                    • Asking for help, clarification, or responding to other answers.

                    • Making statements based on opinion; back them up with references or personal experience.


                    To learn more, see our tips on writing great answers.




                    draft saved


                    draft discarded














                    StackExchange.ready(
                    function () {
                    StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53403306%2fhow-to-batch-convert-sentence-lengths-to-masks-in-pytorch%23new-answer', 'question_page');
                    }
                    );

                    Post as a guest















                    Required, but never shown





















































                    Required, but never shown














                    Required, but never shown












                    Required, but never shown







                    Required, but never shown

































                    Required, but never shown














                    Required, but never shown












                    Required, but never shown







                    Required, but never shown







                    Popular posts from this blog

                    Guess what letter conforming each word

                    Port of Spain

                    Run scheduled task as local user group (not BUILTIN)