Tensorflow에서 선택 사항으로 마킹 입력

Question

defining a custom op in Tensorflow 인 경우 옵션 입력의/개념에 대한 방법이 있습니까? 예를 들어, 입력, 가중치, 이상적으로 선택적으로 편향 용어를 사용하는 융합 된 회선 Op에 대해 작업하고 있습니다. 이상적으로 운영자의 호출자는 편향 용어를 제공하지 않고도 사용할 수 있습니다.

Answer 1

TensorFlow op 정의에 대한 mini-language에는 선택적 Tensor 값 입력이 기본적으로 지원되지 않습니다. 일반적으로 사용되는 몇 가지 대안이있다 : 당신의 연산을 정의 할 때

1. 가 대신 Input()의 Attr()를 사용합니다. 그래프의 구조는 그래프 작성시 고정되어 있으므로 옵션 입력의 유무도 있으므로 컴파일 타임 값으로 정보를 인코딩 할 수 있습니다.

2. op 정의 수준에서 필수 입력을 만들고 해당 인수를 선택적으로 만드는 Python 래퍼 함수를 ​​추가하고 지정되지 않은 경우 기본/전송 값을 추가합니다. 예를 들어, tf.reduce_sum()에 대한 axis 인수는 호출자가 제공하지 않는 경우 채울 선택적 텐서 값 인수입니다.

3. Op 등록은 텐서 목록을 지원하므로 길이가 0 또는 1 인 목록을 사용하여 선택적 입력을 나타낼 수 있습니다.