tensorflow :: operaciones :: BatchMatMulV2

Multiplica porciones de dos tensores en lotes.

Resumen

Multiplica todas las rebanadas de Tensor x y y (cada rebanada se puede ver como un elemento de un lote), y dispone los resultados individuales en un único tensor de salida del mismo tamaño del lote. Cada uno de los cortes individuales se puede adjuntar opcionalmente (adjuntar una matriz significa transponerla y conjugarla) antes de la multiplicación estableciendo el adj_x o adj_y en True , que por defecto es False .

Los tensores de entrada x y y son 2-D o superior con forma [..., r_x, c_x] y [..., r_y, c_y] .

El tensor de salida es 2-D o superior con forma [..., r_o, c_o] , donde:

r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y

Se calcula como:

output[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])

NOTA : BatchMatMulV2 admite la difusión en las dimensiones del lote. Más sobre retransmisiones aquí .

Argumentos:

• alcance: un objeto de alcance
• x: 2-D o superior con forma [..., r_x, c_x] .
• y: 2-D o superior con forma [..., r_y, c_y] .

Atributos opcionales (consulte Attrs ):

• adj_x: si es True , adjunte las porciones de x . El valor predeterminado es False .
• adj_y: si es True , adjunta las secciones de y . El valor predeterminado es False .

Devoluciones:

• Output : 3-D o superior con forma [..., r_o, c_o]

Atributos públicos

Operation operation

::tensorflow::Output output

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y
)

 BatchMatMulV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input y,
  const BatchMatMulV2::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs AdjX(
  bool x
)

Attrs AdjY(
  bool x
)