Batting coach Craig McMillan says that's all they need, despite the quick turnaround from New Zealand's 2-0 Test series win against Pakistan.

"It's one of the greatest challenges for the modern-day cricket player, changing from Test cricket to one-day to T20 cricket, just backwards and forwards," McMillan said.

"More more so than anything, it's probably a mental shift."

New Zealand backed up their first Test eight-wicket victory with a 138-run win in the second Test which ended in Hamilton on Tuesday.

McMillan says the Black Caps' first net session in Australia will be crucial.

"It's about hitting the ball hard, finding some freedom and perhaps playing a few extra shots that you normally put away for Test cricket," he said before the team left for Australia on Wednesday.

"That's that balance and fine line you're looking for, and we need to find that over the two days training before we head into Sunday's match.

"It's about players finding that freedom you need in one-day cricket because quite often that freedom can be lost in Test cricket."

And while he expects the quick change in format will be challenging, McMillan says the Black Caps will take an extra boost in confidence from their first Test series win over Pakistan since 1985.

"It gives you a sense of momentum, which can flow from format to format - it gives you confidence, and it gives you a sense of positivity.

The Black Caps are taking a relatively inexperienced squad across the Tasman, with only Tim Southee, Martin Guptill, Kane Williamson, Trent Boult and Matt Henry remaining from the team which lost last year's World Cup final to Australia.

But McMillan says the current squad boasts some in-form players who impressed against Pakistan, including Test debutant Colin de Grandhomme.

The 30-year-old right-hander grabbed a six-wicket bag in the first Test and finished the series with figures of 9-110.

"It's still up for discussion in terms of the side we're going to pick, but you have to say his performances in those two Test matches were excellent.

"He made the most of his opportunities with both bat and ball, and I think he's the sort of dynamic player you want in your one-day side."