Thanks. That works, I only needed to reshape my input_var into shape seq length X batch X 1
input_var
seq length X batch X 1