Arg(BOOL, "retLastH") {
            description = "indicates whether to return output at last time step only,\n" +
                    " in this case shape would be [bS, nOut] (exact shape depends on dataFormat argument)"; defaultValue = false
        }