Source code for mindspore.nn.probability.bijector.power_transform

# Copyright 2020 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
"""PowerTransform Bijector"""
from mindspore.ops import operations as P
from mindspore.ops import functional as F
from ..distribution._utils.utils import check_greater_equal_zero
from ..distribution._utils.custom_ops import exp_generic, log_generic
from .bijector import Bijector


[docs]class PowerTransform(Bijector): r""" PowerTransform Bijector. This Bijector performs the operation: .. math:: Y = g(X) = (1 + X * c)^{1 / c}, X >= -1 / c where c >= 0 is the power. The power transform maps inputs from `[-1/c, inf]` to `[0, inf]`. This Bijector is equivalent to the :class:`mindspore.nn.probability.bijector.Exp` bijector when `c=0`. Args: power (float, list, numpy.ndarray, Tensor): The scale factor. Default: ``0`` . name (str): The name of the bijector. Default: ``'PowerTransform'`` . Note: The dtype of `power` must be float. Raises: ValueError: When `power` is less than 0 or is not known statically. TypeError: When the dtype of `power` is not float. Supported Platforms: ``Ascend`` ``GPU`` Examples: >>> import mindspore >>> import mindspore.nn as nn >>> import mindspore.nn.probability.bijector as msb >>> from mindspore import Tensor >>> # To initialize a PowerTransform bijector of power 0.5. >>> powertransform = msb.PowerTransform(0.5) >>> value = Tensor([1, 2, 3], dtype=mindspore.float32) >>> ans1 = powertransform.forward(value) >>> print(ans1.shape) (3,) >>> ans2 = powertransform.inverse(value) >>> print(ans2.shape) (3,) >>> ans3 = powertransform.forward_log_jacobian(value) >>> print(ans3.shape) (3,) >>> ans4 = powertransform.inverse_log_jacobian(value) >>> print(ans4.shape) (3,) """ def __init__(self, power=0., name='PowerTransform'): param = dict(locals()) param['param_dict'] = {'power': power} super(PowerTransform, self).__init__(name=name, param=param) self._power = self._add_parameter(power, 'power') check_greater_equal_zero(self._power, 'Power') self.pow = P.Pow() self.dtypeop = P.DType() self.cast = P.Cast() self.equal_base = P.Equal() self.exp = exp_generic self.expm1 = P.Expm1() self.log = log_generic self.log1p = P.Log1p() self.select_base = P.Select() self.shape = P.Shape() @property def power(self): """ Return the power parameter of the bijector. Returns: Tensor, the power parameter of the bijector. """ return self._power def extend_repr(self): """Display instance object as string.""" if self.is_scalar_batch: str_info = 'power = {}'.format(self.power) else: str_info = 'batch_shape = {}'.format(self.batch_shape) return str_info def _forward(self, x): """ Evaluate the forward mapping. """ x = self._check_value_dtype(x) power_local = self.cast_param_by_value(x, self.power) # broad cast the value of x and power ones = F.fill(self.dtypeop(power_local), self.shape(x + power_local), 1.) power_local = power_local * ones x = x * ones safe_power = self.select_base( self.equal_base(power_local, P.ZerosLike()(power_local)), ones, power_local) forward_v = self.select_base( self.equal_base(power_local, P.ZerosLike()(power_local)), self.exp(x), self.exp(self.log1p(x * safe_power) / safe_power)) return forward_v def _inverse(self, y): """ Evaluate the inverse mapping. """ y = self._check_value_dtype(y) power_local = self.cast_param_by_value(y, self.power) # broad cast the value of x and power ones = F.fill(self.dtypeop(power_local), self.shape(y + power_local), 1.) power_local = power_local * ones y = y * ones safe_power = self.select_base( self.equal_base(power_local, P.ZerosLike()(power_local)), ones, power_local) inverse_v = self.select_base( self.equal_base(power_local, P.ZerosLike()(power_local)), self.log(y), self.expm1(self.log(y) * safe_power) / safe_power) return inverse_v def _forward_log_jacobian(self, x): r""" .. math: if c == 0: f(x) = e^x f'(x) = e^x \log(f'(x)) = \log(e^x) = x else: f(x) = e^\frac{\log(xc + 1)}{c} f'(x) = e^\frac{\log(xc + 1)}{c} * \frac{1}{xc + 1} \log(f'(x)) = (\frac{1}{c} - 1) * \log(xc + 1) """ x = self._check_value_dtype(x) power_local = self.cast_param_by_value(x, self.power) # broad cast the value of x and power ones = F.fill(self.dtypeop(power_local), self.shape(x + power_local), 1.) power_local = power_local * ones x = x * ones forward_log_j = self.select_base( self.equal_base(power_local, P.ZerosLike()(power_local)), x, (1. / power_local - 1) * self.log1p(x * power_local)) return forward_log_j def _inverse_log_jacobian(self, y): r""" .. math: if c == 0: f(x) = \log(x) f'(x) = \frac{1}{x} \log(f'(x)) = \log(\frac{1}{x}) = -\log(x) else: f(x) = \frac{e^\log(y)*c + 1}{c} f'(x) = \frac{e^c\log(y)}{y} \log(f'(x)) = \log(\frac{e^c\log(y)}{y}) = (c-1) * \log(y) """ y = self._check_value_dtype(y) power_local = self.cast_param_by_value(y, self.power) inverse_log_j = (power_local - 1) * self.log(y) return inverse_log_j