Python Tensor.dataの例

コード例 #1

    def test_reassign_data_wrong_shape(self) -> None:
        tensor = Tensor([1, 2, 3])

        with pytest.raises(RuntimeError):
            tensor.data = np.array([3])

コード例 #2

    def test_reassign_data(self) -> None:
        tensor = Tensor([1, 2, 3])

        tensor.data = np.array([3, 2, 1])
        assert tensor.data.tolist() == [3, 2, 1]