420
420
< li class ="toctree-l2 "> < a class ="reference internal " href ="../../torchvision/transforms.html#transforms-on-pil-image "> Transforms on PIL.Image</ a > </ li >
421
421
< li class ="toctree-l2 "> < a class ="reference internal " href ="../../torchvision/transforms.html#transforms-on-torch-tensor "> Transforms on torch.*Tensor</ a > </ li >
422
422
< li class ="toctree-l2 "> < a class ="reference internal " href ="../../torchvision/transforms.html#conversion-transforms "> Conversion Transforms</ a > </ li >
423
- < li class ="toctree-l2 "> < a class ="reference internal " href ="../../torchvision/transforms.html#generic-transofrms "> Generic Transofrms </ a > </ li >
423
+ < li class ="toctree-l2 "> < a class ="reference internal " href ="../../torchvision/transforms.html#generic-transforms "> Generic Transforms </ a > </ li >
424
424
</ ul >
425
425
</ li >
426
426
< li class ="toctree-l1 "> < a class ="reference internal " href ="../../torchvision/utils.html "> torchvision.utils</ a > </ li >
@@ -506,7 +506,7 @@ <h1>Source code for torch._utils</h1><div class="highlight"><pre>
506
506
< span class ="sd "> Otherwise, the argument has no effect.</ span >
507
507
< span class ="sd "> """</ span >
508
508
< 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 >
509
- < 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 >
509
+ < 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 >
510
510
511
511
< 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 >
512
512
< 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 >
@@ -515,7 +515,7 @@ <h1>Source code for torch._utils</h1><div class="highlight"><pre>
515
515
< span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> is_sparse</ span > < span class ="p "> :</ span >
516
516
< 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 >
517
517
< 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 >
518
- < 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 >
518
+ < 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 >
519
519
< 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 >
520
520
< 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 >
521
521
< 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 >
@@ -546,18 +546,18 @@ <h1>Source code for torch._utils</h1><div class="highlight"><pre>
546
546
< span class ="n "> device</ span > < span class ="o "> =</ span > < span class ="o "> -</ span > < span class ="mi "> 1</ span >
547
547
< 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 >
548
548
< span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> is_sparse</ span > < span class ="p "> :</ span >
549
- < 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 >
549
+ < 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 >
550
550
< 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 >
551
551
< 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 >
552
552
< 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 >
553
553
< span class ="k "> else</ span > < span class ="p "> :</ span >
554
- < 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 >
554
+ < 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 >
555
555
< 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 >
556
556
557
557
558
558
< span class ="k "> def</ span > < span class ="nf "> _rebuild_tensor</ span > < span class ="p "> (</ span > < span class ="n "> storage</ span > < span class ="p "> ,</ span > < span class ="n "> storage_offset</ span > < span class ="p "> ,</ span > < span class ="n "> size</ span > < span class ="p "> ,</ span > < span class ="n "> stride</ span > < span class ="p "> ):</ span >
559
- < span class ="n "> class_name</ span > < span class ="o "> =</ span > < span class ="n "> storage</ span > < span class ="o "> .</ span > < span class ="vm "> __class__</ span > < span class ="o "> .</ span > < span class ="vm "> __name__</ span > < span class ="o "> .</ span > < span class ="n "> replace</ span > < span class ="p "> (</ span > < span class ="s1 "> 'Storage'</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'Tensor'</ span > < span class ="p "> )</ span >
560
- < span class ="n "> module</ span > < span class ="o "> =</ span > < span class ="n "> importlib</ span > < span class ="o "> .</ span > < span class ="n "> import_module</ span > < span class ="p "> (</ span > < span class ="n "> storage</ span > < span class ="o "> .</ span > < span class ="vm "> __module__</ span > < span class ="p "> )</ span >
559
+ < span class ="n "> class_name</ span > < span class ="o "> =</ span > < span class ="n "> storage</ span > < span class ="o "> .</ span > < span class ="n "> __class__</ span > < span class ="o "> .</ span > < span class ="n "> __name__</ span > < span class ="o "> .</ span > < span class ="n "> replace</ span > < span class ="p "> (</ span > < span class ="s1 "> 'Storage'</ span > < span class ="p "> ,</ span > < span class ="s1 "> 'Tensor'</ span > < span class ="p "> )</ span >
560
+ < span class ="n "> module</ span > < span class ="o "> =</ span > < span class ="n "> importlib</ span > < span class ="o "> .</ span > < span class ="n "> import_module</ span > < span class ="p "> (</ span > < span class ="n "> storage</ span > < span class ="o "> .</ span > < span class ="n "> __module__</ span > < span class ="p "> )</ span >
561
561
< span class ="n "> tensor_class</ span > < span class ="o "> =</ span > < span class ="nb "> getattr</ span > < span class ="p "> (</ span > < span class ="n "> module</ span > < span class ="p "> ,</ span > < span class ="n "> class_name</ span > < span class ="p "> )</ span >
562
562
< span class ="k "> return</ span > < span class ="n "> tensor_class</ span > < span class ="p "> ()</ span > < span class ="o "> .</ span > < span class ="n "> set_</ span > < span class ="p "> (</ span > < span class ="n "> storage</ span > < span class ="p "> ,</ span > < span class ="n "> storage_offset</ span > < span class ="p "> ,</ span > < span class ="n "> size</ span > < span class ="p "> ,</ span > < span class ="n "> stride</ span > < span class ="p "> )</ span >
563
563
0 commit comments