tensorflow_probability/python/distributions/hidden_markov_model.py
def _vector_matrix(vs, ms):
  """Multiply tensor of vectors by matrices."""

  return tf.reduce_sum(input_tensor=vs[..., tf.newaxis] * ms, axis=-2)
def _vector_matrix(vs, ms):
  return tf.reduce_sum(input_tensor=vs[..., tf.newaxis] * ms, axis=-2)
def _vector_matrix ( vs , ms ) : return tf . reduce_sum ( input_tensor = vs [ ... , tf . newaxis ] * ms , axis = - 2 )
Multiply tensor of vectors by matrices.
Multiply tensor of vectors by matrices .