fix plotting with > 1-d tensors in tutorials with py3.9 (#2013)

Summary:
Pull Request resolved: #2013

see title. This is causing a failure in the tutorials on  #2011

Reviewed By: Balandat

Differential Revision: D49382057

fbshipit-source-id: 2b88bfeea442e6c1b3d521ee97b4631ff2c77d60

Author: sdaulton