コード例 #1
0
ファイル: test_initializer.py プロジェクト: zimaxeg/mindspore
def test_init_zero():
    tensor = init.initializer(init.Zero(), [2, 2], ms.float32)
    _check_value(tensor, 0, 0)
コード例 #2
0
ファイル: test_initializer.py プロジェクト: zimaxeg/mindspore
def test_init_zero_default_dtype():
    tensor = init.initializer(init.Zero(), [2, 2])
    assert tensor.dtype() == ms.float32
    _check_value(tensor, 0, 0)