ML 커뮤니티 데이는 11월 9일입니다! TensorFlow, JAX에서 업데이트를 우리와 함께, 더 자세히 알아보기

tensorflow :: ops :: 스위치

#include <control_flow_ops.h>

pred 의해 결정된 출력 포트로 data 를 전달 data .

요약

pred 가 true이면 data 입력이 output_true 로 전달 output_true . 그렇지 않으면 데이터가 output_false 로 이동합니다.

RefSwitchMerge 참고하십시오.

인수 :

  • 범위 : 범위 개체
  • data : 적절한 출력으로 전달할 텐서입니다.
  • pred : 데이터를 수신 할 출력 포트를 지정하는 스칼라.

보고:

  • Output output_false : pred 가 false이면 데이터가이 출력으로 전달됩니다.
  • Output output_true : pred 가 true이면 데이터가이 출력으로 전달됩니다.

생성자와 소멸자

Switch (const :: tensorflow::Scope & scope, :: tensorflow::Input data, :: tensorflow::Input pred)

공용 속성

operation
output_false
output_true

공용 속성

조작

Operation operation

output_false

::tensorflow::Output output_false

output_true

::tensorflow::Output output_true

공공 기능

스위치

 Switch(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input pred
)