title: Tensorflow batch sparse multiply
body: <p>I would like to multiply a sparse tensor by a dense tensor but do so within a batch. </p>
  
  <p>For example I have a sparse tensor with the corresponding dense shape of (20,65536,65536) where 20 is the batch size. I would like to multiply each (65536,65536) in the batch with the corresponding (65536x1) from a tensor shape (20,65536) which has a dense representation. <code>tf.sparse_tensor_dense_matmul</code> only accepts a rank 2 sparse tensor. Is there a way to perform this over a batch?</p>
  
  <p>I would like to avoid converting the sparse matrix to a dense matrix if possible due to memory constraints.</p>