|
- # -*- coding: utf-8 -*-
- # MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
- #
- # Copyright (c) 2014-2020 Megvii Inc. All rights reserved.
- #
- # Unless required by applicable law or agreed to in writing,
- # software distributed under the License is distributed on an
- # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- import numpy as np
-
- import megengine.functional as F
- from megengine import tensor
-
- from megengine.test import assertTensorClose
-
-
- def test_abs():
- assertTensorClose(
- F.abs(tensor([-3., -4., -5.])).numpy(),
- np.abs(np.array([-3., -4., -5.], dtype=np.float32)))
-
- assertTensorClose(F.abs(-3.), np.abs(np.float32(-3.)))
-
-
- def test_multiply():
- assertTensorClose(F.multiply(-3., -4.),
- np.multiply(np.float32(-3.), np.float32(-4.)))
-
- assertTensorClose(
- F.multiply(tensor([3., 4.]), 4.).numpy(),
- np.multiply(np.array([3., 4.], dtype=np.float32), 4.))
-
- assertTensorClose(
- F.multiply(4., tensor([3., 4.])).numpy(),
- np.multiply(4., np.array([3., 4.], dtype=np.float32)))
-
- assertTensorClose(
- F.multiply(tensor([3., 4.]), tensor([3., 4.])).numpy(),
- np.multiply(np.array([3., 4.], dtype=np.float32),
- np.array([3., 4.], dtype=np.float32)))
|
