Esta página foi traduzida pela API Cloud Translation.
Switch to English

tf.register_tensor_conversion_function

TensorFlow 1 versão Ver fonte no GitHub

Registra uma função para converter objetos de base_type para Tensor .

A função de conversão deve ter a seguinte assinatura:

     def conversion_func(value, dtype=None, name=None, as_ref=False):
      # ...
 

Ele deve retornar um Tensor com o dado dtype se especificado. Se a função de conversão cria um novo Tensor , ele deve usar o dado name se especificado. Todas as exceções serão propagadas para o chamador.

A função de conversão pode retornar NotImplemented para algumas entradas. Neste caso, o processo de conversão vai continuar a tentar funções de conversão subsequentes.

Se as_ref é verdade, a função deve retornar um Tensor de referência, como uma Variable .

base_type O tipo base ou tupla de tipos de base para todos os objetos que conversion_func aceita.
conversion_func Uma função que converte instâncias de base_type para Tensor .
priority inteiro opcional que indica a prioridade para aplicar esta função de conversão. funções de conversão com valores de prioridade menores executar anteriores à conversão funções com valores de prioridade maiores. O padrão é 100.

TypeError Se os argumentos não têm o tipo apropriado.