Row-wise tensor product using TensorFlow
tf_row_tensor(a, b, ...)
a
b
tensor
...
arguments passed to TensorFlow layer
a TensorFlow layer