コード例 #1
0
ファイル: test_math.py プロジェクト: hvy/chainer
def test_truediv_scalar(scalar, device, shape, numeric_dtype):
    x_np = array_utils.create_dummy_ndarray(numpy, shape, numeric_dtype)
    expected = x_np / scalar

    x = chainerx.array(x_np)
    scalar_chx = chainerx.Scalar(scalar)
    chainerx.testing.assert_array_equal_ex(
        x / scalar, expected, strides_check=False)
    chainerx.testing.assert_array_equal_ex(
        x / scalar_chx, expected, strides_check=False)
    chainerx.testing.assert_array_equal_ex(
        chainerx.divide(x, scalar), expected, strides_check=False)
    chainerx.testing.assert_array_equal_ex(
        chainerx.divide(x, scalar_chx), expected, strides_check=False)
コード例 #2
0
ファイル: test_math.py プロジェクト: knok/chainer
def test_truediv_scalar(scalar, device, shape, numeric_dtype):
    x_np = array_utils.create_dummy_ndarray(numpy, shape, numeric_dtype)
    expected = x_np / scalar

    x = chainerx.array(x_np)
    scalar_chx = chainerx.Scalar(scalar, numeric_dtype)
    chainerx.testing.assert_array_equal_ex(
        x / scalar, expected, strides_check=False)
    chainerx.testing.assert_array_equal_ex(
        x / scalar_chx, expected, strides_check=False)
    chainerx.testing.assert_array_equal_ex(
        chainerx.divide(x, scalar), expected, strides_check=False)
    chainerx.testing.assert_array_equal_ex(
        chainerx.divide(x, scalar_chx), expected, strides_check=False)
コード例 #3
0
def test_truediv_scalar(scalar, device, shape, numeric_dtype):
    x_np = array_utils.create_dummy_ndarray(numpy, shape, numeric_dtype)
    if 'int' in numeric_dtype:
        # NumPy does not support itruediv to integer arrays.
        expected = (x_np / scalar).astype(numeric_dtype)
    else:
        expected = x_np / scalar

    x = chainerx.array(x_np)
    scalar_chx = chainerx.Scalar(scalar, numeric_dtype)
    chainerx.testing.assert_array_equal_ex(x / scalar, expected)
    chainerx.testing.assert_array_equal_ex(x / scalar_chx, expected)
    chainerx.testing.assert_array_equal_ex(chainerx.divide(x, scalar),
                                           expected)
    chainerx.testing.assert_array_equal_ex(chainerx.divide(x, scalar_chx),
                                           expected)
コード例 #4
0
ファイル: test_math.py プロジェクト: jnishi/chainer
def test_truediv_scalar(scalar, device, shape, numeric_dtype):
    x_np = array_utils.create_dummy_ndarray(numpy, shape, numeric_dtype)
    if 'int' in numeric_dtype:
        # NumPy does not support itruediv to integer arrays.
        expected = (x_np / scalar).astype(numeric_dtype)
    else:
        expected = x_np / scalar

    x = chainerx.array(x_np)
    scalar_chx = chainerx.Scalar(scalar, numeric_dtype)
    chainerx.testing.assert_array_equal_ex(x / scalar, expected)
    chainerx.testing.assert_array_equal_ex(x / scalar_chx, expected)
    chainerx.testing.assert_array_equal_ex(
        chainerx.divide(x, scalar), expected)
    chainerx.testing.assert_array_equal_ex(
        chainerx.divide(x, scalar_chx), expected)