/usr/local/lib64/python3.6/site-packages/torch/__pycache__
Edit: /usr/local/lib64/python3.6/site-packages/torch/__pycache__/storage.cpython-36.pyc (8334B)
3
Eg @ sh d dl Z d dlZddlmZmZ d dlmZmZmZ edddZ G dd de
Z d d
Z ee _
ee _dS ) N )_type_cuda)AnyTypeVarTypeT _StorageBase)boundc s e Zd ZU edZedZedd Ze dddZ
dd Z e e d d
d Z
e dd d
ZdSeee dddZdTe dddZe dddZe dddZdd Zdd Zeee e e d d d Zeee e e d d d Zd!d" Zd#d$ Zd%d& Z d'd( Z d)d* Z d+d, Z fd-d.Z d/d0 Z!d1d2 Z"d3d4 Z#d5d6 Z$d7d8 Z%d9d: Z&d;d< Z'd=d> Z d?d@ Z(dAdB Z)dCdD Z*dEdF ZdGdH Z+dIdJ Z,dKdL Z-dMdN Z.dOdP Z/edQdR Z0 Z1S )Ur Fc O s d S )N )selfargskwargsr r 9/usr/local/lib64/python3.6/site-packages/torch/storage.py__init__
s z_StorageBase.__init__)returnc C s d S )Nr )r r r r __len__ s z_StorageBase.__len__c C s d S )Nr )r idxr r r __getitem__ s z_StorageBase.__getitem__)sourcer c C s d S )Nr )r r r r r copy_ s z_StorageBase.copy_c C s d S )Nr )r r r r size s z_StorageBase.sizeN)dtype non_blockingr c C s d S )Nr )r r r r r r type s z_StorageBase.typec K s d S )Nr )r devicer r r r r cuda s z_StorageBase.cudac C s d S )Nr )r r r r element_size s z_StorageBase.element_sizec C s d S )Nr )r r r r
get_device s z_StorageBase.get_devicec C s d S )Nr )r r r r _share_filename_ s z _StorageBase._share_filename_c C s d S )Nr )r r r r
_share_fd_ s z_StorageBase._share_fd_)clsr r c C s d S )Nr )r! r r r r _new_using_filename s z _StorageBase._new_using_filenamec C s d S )Nr )r! r r r r
_new_using_fd s z_StorageBase._new_using_fdc sD ddj fddtt D }|dtj dt d S )N z
c 3 s | ]}t | V qd S )N)str).0i)r r r
s z'_StorageBase.__str__..z
[z of size ])joinrangelentorchtypename)r contentr )r r __str__ s $z_StorageBase.__str__c C s t | S )N)r% )r r r r __repr__# s z_StorageBase.__repr__c s t t fddt j S )Nc s | S )Nr )r' )r r r ' s z'_StorageBase.__iter__..)itermapr+ r )r r )r r __iter__& s z_StorageBase.__iter__c C s | j S )N)clone)r r r r __copy__) s z_StorageBase.__copy__c C s6 |j di }| j|kr || j S | j }||| j< |S )Nr- )
setdefault_cdatar6 )r memoZ new_storager r r __deepcopy__, s
z_StorageBase.__deepcopy__c C s& t j }tj| |dd t|j ffS )NF)Z _use_new_zipfile_serialization)ioBytesIOr- save_load_from_bytesgetvalue)r br r r
__reduce__4 s z_StorageBase.__reduce__c s t t| j | j | j S )N)superr
__sizeof__r r )r ) __class__r r rD 9 s z_StorageBase.__sizeof__c C s@ | j r| j nd}tjj| t| | j j| S Q R X dS )z Returns a copy of this storager N)is_cudar r- r r r r r )r r r r r r6 < s z_StorageBase.clonec C s t | S )z6Returns a list containing the elements of this storage)list)r r r r tolistB s z_StorageBase.tolistc C s | j tt| jjS )zAReturns a CPU copy of this storage if it's not already on the CPU)r getattrr- rE __name__)r r r r cpuF s z_StorageBase.cpuc C s | j t | jd S )z!Casts this storage to double typez.DoubleStorage)r
__module__)r r r r doubleJ s z_StorageBase.doublec C s | j t | jd S )z Casts this storage to float typez
.FloatStorage)r rM )r r r r floatN s z_StorageBase.floatc C s | j t | jd S )z Casts this storage to half typez .HalfStorage)r rM )r r r r halfR s z_StorageBase.halfc C s | j t | jd S )z Casts this storage to long typez .LongStorage)r rM )r r r r longV s z_StorageBase.longc C s | j t | jd S )z Casts this storage to int typez .IntStorage)r rM )r r r r intZ s z_StorageBase.intc C s | j t | jd S )z Casts this storage to short typez
.ShortStorage)r rM )r r r r short^ s z_StorageBase.shortc C s | j t | jd S )z Casts this storage to char typez .CharStorage)r rM )r r r r charb s z_StorageBase.charc C s | j t | jd S )z Casts this storage to byte typez .ByteStorage)r rM )r r r r bytef s z_StorageBase.bytec C s | j t | jd S )z Casts this storage to bool typez .BoolStorage)r rM )r r r r boolj s z_StorageBase.boolc C s | j t | jd S )z#Casts this storage to bfloat16 typez.BFloat16Storage)r rM )r r r r bfloat16n s z_StorageBase.bfloat16c C s | j t | jd S )z)Casts this storage to complex double typez.ComplexDoubleStorage)r rM )r r r r complex_doubler s z_StorageBase.complex_doublec C s | j t | jd S )z(Casts this storage to complex float typez.ComplexFloatStorage)r rM )r r r r
complex_floatv s z_StorageBase.complex_floatc C sF | j rtd| j dddl}|jj }t| | j |dj| S )z@Copies the storage to pinned memory, if it's not already pinned.z cannot pin 'z ' only CPU memory can be pinnedr N) allocator)rG TypeErrorr Z
torch.cudar Z_host_allocatorr r )r r- rZ r r r
pin_memoryz s
z_StorageBase.pin_memoryc C s4 ddl m} | jrn | dkr(| j n| j | S )a Moves the storage to shared memory.
This is a no-op for storages already in shared memory and for CUDA
storages, which do not need to be moved for sharing across processes.
Storages in shared memory cannot be resized.
Returns: self
r )get_sharing_strategy file_system)torch.multiprocessingr] rG r r )r r] r r r
share_memory_ s
z_StorageBase.share_memory_c C s< ddl m} | jr| |S | dkr.| j|S | j|S dS )z>Creates a new storage in shared memory with the same data typer )r] r^ N)r_ r] rG r" r# )r! r r] r r r _new_shared s
z_StorageBase._new_shared)NF)NF)2rK rM __qualname__r r9 rG rV Z is_sparser rR r r r r r r% r r r r r r classmethodr r" r# r0 r1 r5 r7 r; rB rD r6 rI rL rN rO rP rQ rS rT rU rW rX rY r\ r` ra
__classcell__r r )rE r r sV
c C s t jtj| S )N)r- loadr< r= )rA r r r r? s r? )r< r- _utilsr r typingr r r r objectr r? r r r r r r s