diff options
| author | Vishwak Srinivasan <cs15btech11043@iith.ac.in> | 2017-12-28 15:26:03 +0530 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-12-28 18:56:03 +0900 |
| commit | e519ef53378c15c8d6ba582aba454b2e1ff6169e (patch) | |
| tree | 7851d0dc4bd8fb64088020363e6b1a4508f99a1f /aten/doc | |
| parent | d859c3c7cc129dc872871981f28c068c3a4ba494 (diff) | |
| download | pytorch-e519ef53378c15c8d6ba582aba454b2e1ff6169e.tar.gz pytorch-e519ef53378c15c8d6ba582aba454b2e1ff6169e.tar.bz2 pytorch-e519ef53378c15c8d6ba582aba454b2e1ff6169e.zip | |
Adding torch.expm1() and its inplace function (#4350)
Diffstat (limited to 'aten/doc')
| -rw-r--r-- | aten/doc/Functions.h | 8 | ||||
| -rw-r--r-- | aten/doc/Tensor.h | 2 | ||||
| -rw-r--r-- | aten/doc/Type.h | 3 |
3 files changed, 13 insertions, 0 deletions
diff --git a/aten/doc/Functions.h b/aten/doc/Functions.h index fef2e8b47e..b262dc4598 100644 --- a/aten/doc/Functions.h +++ b/aten/doc/Functions.h @@ -129,6 +129,8 @@ static inline Tensor & lgamma_out(Tensor & result, const Tensor & self); static inline Tensor lgamma(const Tensor & self); static inline Tensor & exp_out(Tensor & result, const Tensor & self); static inline Tensor exp(const Tensor & self); +static inline Tensor & expm1_out(Tensor & result, const Tensor & self); +static inline Tensor expm1(const Tensor & self); static inline Tensor & cos_out(Tensor & result, const Tensor & self); static inline Tensor cos(const Tensor & self); static inline Tensor & acos_out(Tensor & result, const Tensor & self); @@ -909,6 +911,12 @@ static inline Tensor & exp_out(Tensor & result, const Tensor & self) { static inline Tensor exp(const Tensor & self) { return infer_type(self).exp(self); } +static inline Tensor & expm1_out(const Tensor & result, const Tensor & self) { + return infer_type(self).exp_out(result, self); +} +static inline Tensor expm1(const Tensor & self) { + return infer_type(self).exp(self); +} static inline Tensor & cos_out(Tensor & result, const Tensor & self) { return infer_type(self).cos_out(result, self); } diff --git a/aten/doc/Tensor.h b/aten/doc/Tensor.h index c98f392623..7cca1aaf09 100644 --- a/aten/doc/Tensor.h +++ b/aten/doc/Tensor.h @@ -244,6 +244,8 @@ struct Tensor : public detail::TensorBase { Tensor & lgamma_(); Tensor & exp_(); Tensor exp() const; + Tensor & expm1_(); + Tensor expm1() const; Tensor & cos_(); Tensor cos() const; Tensor & acos_(); diff --git a/aten/doc/Type.h b/aten/doc/Type.h index c14fd2290b..d9cc645594 100644 --- a/aten/doc/Type.h +++ b/aten/doc/Type.h @@ -321,6 +321,9 @@ struct AT_API Type { virtual Tensor & m_exp_(Tensor & self) const; virtual Tensor & exp_out(Tensor & result, const Tensor & self) const; virtual Tensor exp(const Tensor & self) const; + virtual Tensor & m_expm1_(Tensor & self) const; + virtual Tensor & expm1_out(Tensor & result, const Tensor & self) const; + virtual Tensor expm1(const Tensor & self) const; virtual Tensor & m_cos_(Tensor & self) const; virtual Tensor & cos_out(Tensor & result, const Tensor & self) const; virtual Tensor cos(const Tensor & self) const; |