torch.set_printoptions()

torch.set_printoptions(precision=None, threshold=None, edgeitems=None, linewidth=None, profile=None) precision是每一個元素的輸出精度,默認是八位; threshold是輸出時的閾值,當tensor中元素的個數大於該值時,進行縮略輸出,默認時1000; edgeitems是輸出的維度,默認是
本站公眾號
   歡迎關注本站公眾號,獲取更多信息