Tensorflow에 새 Op 추가 : 모양 함수

Question

새로운 방법 (이 방법을 사용하여 : https://www.tensorflow.org/how_tos/adding_an_op/)을 추가하려고합니다. 예와 같이 단순한 Op에는 문제가 없지만 더 복잡한 Op를 추가해야합니다. 그것은 2 개의 입력과 입력의 행렬 곱셈과 같은 모양을 갖는 출력을 가져야합니다. 이 경우 어떻게 모양 함수를 작성할 수 있습니까? 출력을 적절한 모양으로 할당하려면 어떻게해야합니까? 미리 감사드립니다.

Answer 1

질문에 부분적으로 대답하십시오. 나는 아직도 형상 함수를 작성하는 방법을 모르겠지만, 나는이 같은 검사를 썼다 :

const Tensor& input1 = context->input(0); 
const Tensor& input2 = context->input(1); 
TensorShape sh1 = input1.shape(); 
TensorShape sh2 = input2.shape(); 
OP_REQUIRES(context, sh1.dim_size(1)==sh2.dim_size(0), 
errors::InvalidArgument("Can't multiplicate!")); 

할당 :

sh1.RemoveDim(1); 
sh2.RemoveDim(0); 
sh1.AppendShape(sh2); 
OP_REQUIRES_OK(context, context->allocate_output(0, sh1, &output_tensor)); 

을하지만 자전거를 개혁하고있어처럼 보인다. 더 쉬운 방법일까요?