Rewrote the api for imrange and grid samples 
Rewrote the api for imrange and grid samples 

New api is to output a pytorch tensor which is a lot easier to manipulate than matplotlib subplots
probml/pyprobml,probml/pyprobml,probml/pyprobml,probml/pyprobml