在Pytorch中绘制torch.tensor:无效尺寸错误

时间:2019-04-22 07:22:02

标签: matplotlib pytorch

我有一个 var dashboard = new google.visualization.Dashboard(chartSliderPos); var rangeSlider = new google.visualization.ControlWrapper({ 'controlType': 'DateRangeFilter', 'containerId': chartSliderPos, 'options': { 'filterColumnLabel': 'Time', 'ui': { 'step': 'hour' } } }); // Create our data table out of JSON data loaded from server. var data = new google.visualization.DataTable(); data.addColumn("date", "Time"); data.addColumn("number", "Value"); data.addRows(resultArray); // Instantiate and draw our chart, passing in some options. var chart = new google.visualization.ChartWrapper({ 'chartType': 'SteppedAreaChart', 'containerId': pos, 'options': { 'width': 750, 'height': 300, 'explorer': {}, 'legend': 'none' } }); dashboard.bind(rangeSlider, chart); //callback for slider manipulation chartsArray.push( function(a, b) { rangeSlider.setState({'lowValue': a, 'highValue': b}); rangeSlider.draw(); }); dashboard.draw(data); 对象,它的形状为torch.Tensor

我尝试过类似的

torch.Size([9, 1, 28, 28]))

但这给了我:for digit in range(10): similar_img = create_interpolates(/*something...*/) plt.figure(figsize=(10,10)) plt.imshow(similar_img.detach().numpy())

有帮助吗?

1 个答案:

答案 0 :(得分:0)

请尝试以下代码段。

for digit in range(10):
    similar_img = create_interpolates(/*something...*/)
    plt.figure(figsize=(10,10))
    plt.imshow(similar_img[digit].numpy().squeeze())

谢谢