348
348
</ li >
349
349
< li class ="toctree-l2 "> < a class ="reference internal " href ="../../nn.html#vision-functions "> Vision functions</ a > < ul >
350
350
< li class ="toctree-l3 "> < a class ="reference internal " href ="../../nn.html#pixel-shuffle "> < span class ="hidden-section "> pixel_shuffle</ span > </ a > </ li >
351
+ < li class ="toctree-l3 "> < a class ="reference internal " href ="../../nn.html#pad "> < span class ="hidden-section "> pad</ span > </ a > </ li >
351
352
</ ul >
352
353
</ li >
353
354
</ ul >
@@ -504,7 +505,7 @@ <h1>Source code for torch._utils</h1><div class="highlight"><pre>
504
505
< span class ="sd "> Otherwise, the argument has no effect.</ span >
505
506
< span class ="sd "> """</ span >
506
507
< span class ="k "> if</ span > < span class ="n "> new_type</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
507
- < span class ="k "> return</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="vm "> __module__</ span > < span class ="o "> +</ span > < span class ="s1 "> '.'</ span > < span class ="o "> +</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="vm "> __class__</ span > < span class ="o "> .</ span > < span class ="vm "> __name__</ span >
508
+ < span class ="k "> return</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> __module__</ span > < span class ="o "> +</ span > < span class ="s1 "> '.'</ span > < span class ="o "> +</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> __class__</ span > < span class ="o "> .</ span > < span class ="n "> __name__</ span >
508
509
509
510
< span class ="k "> if</ span > < span class ="nb "> isinstance</ span > < span class ="p "> (</ span > < span class ="n "> new_type</ span > < span class ="p "> ,</ span > < span class ="nb "> str</ span > < span class ="p "> ):</ span >
510
511
< span class ="n "> new_type</ span > < span class ="o "> =</ span > < span class ="n "> _import_dotted_name</ span > < span class ="p "> (</ span > < span class ="n "> new_type</ span > < span class ="p "> )</ span >
@@ -513,7 +514,7 @@ <h1>Source code for torch._utils</h1><div class="highlight"><pre>
513
514
< span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> is_sparse</ span > < span class ="p "> :</ span >
514
515
< span class ="k "> if</ span > < span class ="ow "> not</ span > < span class ="n "> new_type</ span > < span class ="o "> .</ span > < span class ="n "> is_sparse</ span > < span class ="p "> :</ span >
515
516
< span class ="k "> raise</ span > < span class ="ne "> RuntimeError</ span > < span class ="p "> (</ span > < span class ="s2 "> "Cannot cast sparse tensor to dense tensor"</ span > < span class ="p "> )</ span >
516
- < span class ="n "> new_type_name</ span > < span class ="o "> =</ span > < span class ="n "> new_type</ span > < span class ="o "> .</ span > < span class ="vm "> __module__</ span > < span class ="o "> +</ span > < span class ="s1 "> '.'</ span > < span class ="o "> +</ span > < span class ="n "> new_type</ span > < span class ="o "> .</ span > < span class ="vm "> __name__</ span >
517
+ < span class ="n "> new_type_name</ span > < span class ="o "> =</ span > < span class ="n "> new_type</ span > < span class ="o "> .</ span > < span class ="n "> __module__</ span > < span class ="o "> +</ span > < span class ="s1 "> '.'</ span > < span class ="o "> +</ span > < span class ="n "> new_type</ span > < span class ="o "> .</ span > < span class ="n "> __name__</ span >
517
518
< span class ="n "> new_values_type_name</ span > < span class ="o "> =</ span > < span class ="n "> new_type_name</ span > < span class ="o "> .</ span > < span class ="n "> replace</ span > < span class ="p "> (</ span > < span class ="s1 "> '.sparse'</ span > < span class ="p "> ,</ span > < span class ="s1 "> ''</ span > < span class ="p "> )</ span >
518
519
< span class ="n "> new_values</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> values</ span > < span class ="p "> ()</ span > < span class ="o "> .</ span > < span class ="n "> type</ span > < span class ="p "> (</ span > < span class ="n "> new_values_type_name</ span > < span class ="p "> ,</ span > < span class ="k "> async</ span > < span class ="p "> )</ span >
519
520
< span class ="k "> return</ span > < span class ="n "> new_type</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> indices</ span > < span class ="p "> (),</ span > < span class ="n "> new_values</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> size</ span > < span class ="p "> ())</ span >
@@ -544,12 +545,12 @@ <h1>Source code for torch._utils</h1><div class="highlight"><pre>
544
545
< span class ="n "> device</ span > < span class ="o "> =</ span > < span class ="o "> -</ span > < span class ="mi "> 1</ span >
545
546
< span class ="k "> with</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> cuda</ span > < span class ="o "> .</ span > < span class ="n "> device</ span > < span class ="p "> (</ span > < span class ="n "> device</ span > < span class ="p "> ):</ span >
546
547
< span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> is_sparse</ span > < span class ="p "> :</ span >
547
- < span class ="n "> new_type</ span > < span class ="o "> =</ span > < span class ="nb "> getattr</ span > < span class ="p "> (</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> cuda</ span > < span class ="o "> .</ span > < span class ="n "> sparse</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="vm "> __class__</ span > < span class ="o "> .</ span > < span class ="vm "> __name__</ span > < span class ="p "> )</ span >
548
+ < span class ="n "> new_type</ span > < span class ="o "> =</ span > < span class ="nb "> getattr</ span > < span class ="p "> (</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> cuda</ span > < span class ="o "> .</ span > < span class ="n "> sparse</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> __class__</ span > < span class ="o "> .</ span > < span class ="n "> __name__</ span > < span class ="p "> )</ span >
548
549
< span class ="n "> indices</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> indices</ span > < span class ="p "> ()</ span > < span class ="o "> .</ span > < span class ="n "> cuda</ span > < span class ="p "> (</ span > < span class ="n "> device</ span > < span class ="p "> ,</ span > < span class ="k "> async</ span > < span class ="p "> )</ span >
549
550
< span class ="n "> values</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> values</ span > < span class ="p "> ()</ span > < span class ="o "> .</ span > < span class ="n "> cuda</ span > < span class ="p "> (</ span > < span class ="n "> device</ span > < span class ="p "> ,</ span > < span class ="k "> async</ span > < span class ="p "> )</ span >
550
551
< span class ="k "> return</ span > < span class ="n "> new_type</ span > < span class ="p "> (</ span > < span class ="n "> indices</ span > < span class ="p "> ,</ span > < span class ="n "> values</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> size</ span > < span class ="p "> ())</ span >
551
552
< span class ="k "> else</ span > < span class ="p "> :</ span >
552
- < span class ="n "> new_type</ span > < span class ="o "> =</ span > < span class ="nb "> getattr</ span > < span class ="p "> (</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> cuda</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="vm "> __class__</ span > < span class ="o "> .</ span > < span class ="vm "> __name__</ span > < span class ="p "> )</ span >
553
+ < span class ="n "> new_type</ span > < span class ="o "> =</ span > < span class ="nb "> getattr</ span > < span class ="p "> (</ span > < span class ="n "> torch</ span > < span class ="o "> .</ span > < span class ="n "> cuda</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> __class__</ span > < span class ="o "> .</ span > < span class ="n "> __name__</ span > < span class ="p "> )</ span >
553
554
< span class ="k "> return</ span > < span class ="n "> new_type</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> size</ span > < span class ="p "> ())</ span > < span class ="o "> .</ span > < span class ="n "> copy_</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="k "> async</ span > < span class ="p "> )</ span >
554
555
555
556
0 commit comments