더북(TheBook)

변수의 상태는 다음과 같이 assign 메서드로 수정할 수 있습니다.

코드 3-6 텐서플로 변수에 값 할당하기

>>> v.assign(tf.ones((3, 1)))
array([[1.], 
      [1.], 
      [1.]], dtype=float32)>

변수의 일부 원소에만 적용할 수도 있습니다.

코드 3-7 변수 일부에 값 할당하기

>>> v[0, 0].assign(3.)
array([[3.],
      [1.],
      [1.]], dtype=float32)>

비슷하게 assign_add()assign_sub()은 각각 +=, -=과 동일합니다.

코드 3-8 assign_add( ) 사용하기

>>> v.assign_add(tf.ones((3, 1)))
array([[4.],
      [2.],
      [2.]], dtype=float32)>
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.