tensorflow :: operaciones :: SparseTensorDenseMatMul

#include <sparse_ops.h>

Multiplique SparseTensor (de rango 2) "A" por la matriz densa "B".

Resumen

No se realiza ninguna verificación de validez en los índices de A. Sin embargo, se recomienda el siguiente formato de entrada para un comportamiento óptimo:

if adjoint_a == false: A debe clasificarse en orden lexicográficamente creciente. Utilice SparseReorder si no está seguro. if adjoint_a == true: A debe ordenarse en orden de dimensión creciente 1 (es decir, orden "columna principal" en lugar de orden "fila principal").

Argumentos:

  • alcance: un objeto de alcance
  • a_indices: 2-D. Los indices de la SparseTensor , tamaño [nnz, 2] .
  • a_valores: 1-D. Los values del SparseTensor , tamaño [nnz] .
  • a_forma: 1-D. La shape del SparseTensor , tamaño [2] Vector.
  • b: 2-D. Una matriz densa.

Atributos opcionales (consulte Attrs ):

  • adjoint_a: Usa el adjunto de A en la matriz multiplica. Si A es complejo, se transpone (conj (A)). De lo contrario, se transpone (A).
  • adjoint_b: Usa el adjunto de B en la matriz multiplicar. Si B es complejo, se transpone (conj (B)). De lo contrario, se transpone (B).

Devoluciones:

  • Output : el tensor del producto.

Constructores y Destructores

SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b)
SparseTensorDenseMatMul (const :: tensorflow::Scope & scope, :: tensorflow::Input a_indices, :: tensorflow::Input a_values, :: tensorflow::Input a_shape, :: tensorflow::Input b, const SparseTensorDenseMatMul::Attrs & attrs)

Atributos públicos

operation
product

Funciones publicas

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Funciones estáticas públicas

AdjointA (bool x)
AdjointB (bool x)

Estructuras

tensorflow :: ops :: SparseTensorDenseMatMul :: Attrs

Establecedores de atributos opcionales para SparseTensorDenseMatMul .

Atributos públicos

operación

Operation operation

producto

::tensorflow::Output product

Funciones publicas

SparseTensorDenseMatMul

 SparseTensorDenseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b
)

SparseTensorDenseMatMul

 SparseTensorDenseMatMul(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input a_indices,
  ::tensorflow::Input a_values,
  ::tensorflow::Input a_shape,
  ::tensorflow::Input b,
  const SparseTensorDenseMatMul::Attrs & attrs
)

nodo

::tensorflow::Node * node() const

operador :: tensorflow :: Entrada

 operator::tensorflow::Input() const

operador :: tensorflow :: Salida

 operator::tensorflow::Output() const

Funciones estáticas públicas

AdjointA

Attrs AdjointA(
  bool x
)

AdjointB

Attrs AdjointB(
  bool x
)