U Kc@sLddlmZddlZddlZddlmZGdddeZGdddeZdS) )reduceN)Functionc@s$eZdZeddZeddZdS)TypecCs(t||_|jsdn||_||SN)type input_typeZis_cudaZ get_device input_device)ctxiZ dest_typer D/tmp/pip-unpacked-wheel-gikjz4vx/torch/autograd/_functions/tensor.pyforward s z Type.forwardc CsP|jdkr||jdfStj|j||jdfW5QRSQRXdSr)r rr torchZcudaZdevicer Z grad_outputr r rbackwards z Type.backwardN__name__ __module__ __qualname__ staticmethodrrr r r rrs rc@s$eZdZeddZeddZdS)Resizec Cs||_tdd|d|_||jkr\tddtt||jdtt||||_ |j r| || j |S|r|| j |}|S| j |SdS)NcSs||SNr )xyr r rz Resize.forward..zrequested resize to {} ({} elements in total), but the given tensor has a size of {} ({} elements). autograd's resize can only change the shape of a given tensor, while preserving the number of elements. r)sizesrnumel RuntimeErrorformatjoinmapstrsize input_sizesZ is_quantizedZcopy_ contiguousviewZ is_contiguousnew)r Ztensorrresultr r rrs"  zResize.forwardcCs&||jkst||jdfSr)r AssertionErrorr(r)r'rr r rr0szResize.backwardNrr r r rrs r) functoolsrrZ torch._utilsfunctionrrrr r r rs