|
|
|
@@ -157,6 +157,7 @@ def test_parameter_compute(): |
|
|
|
|
|
|
|
|
|
|
|
def test_scalar_parameter_update(): |
|
|
|
# float |
|
|
|
fp = Parameter(0.5, 'fp') |
|
|
|
fp.default_input = 0.8 |
|
|
|
assert np.array_equal(fp.default_input.asnumpy(), np.array(0.8, np.float32)) |
|
|
|
@@ -167,6 +168,26 @@ def test_scalar_parameter_update(): |
|
|
|
assert np.array_equal(int_.default_input.asnumpy(), np.array(2, np.int32)) |
|
|
|
with pytest.raises(TypeError): |
|
|
|
int_.default_input = 1.2 |
|
|
|
# Tensor |
|
|
|
fp32 = Tensor(0.5, mstype.float32) |
|
|
|
int32 = Tensor(2, mstype.int32) |
|
|
|
fp16 = Tensor(0.6, mstype.float16) |
|
|
|
int16 = Tensor(3, mstype.int16) |
|
|
|
bool_ = Tensor(np.array(True, dtype=np.bool_)) |
|
|
|
# updata_by_tensor |
|
|
|
fp32_p = Parameter(fp32, 'fp32') |
|
|
|
fp32_p.default_input = 0.8 |
|
|
|
fp32_p.default_input = 1 |
|
|
|
fp32_p.default_input = int32 |
|
|
|
fp32_p.default_input = fp32 |
|
|
|
fp32_p.default_input = int16 |
|
|
|
fp32_p.default_input = fp16 |
|
|
|
fp32_p.default_input = bool_ |
|
|
|
|
|
|
|
# updata_by_tensor |
|
|
|
fp16_p = Parameter(fp16, 'fp16') |
|
|
|
with pytest.raises(TypeError): |
|
|
|
fp16_p.default_input = fp32 |
|
|
|
|
|
|
|
|
|
|
|
def test_parameter_lazy_init(): |
|
|
|
|