Solution of ellipsis when pytorch outputs tensor (output tensor completely)

Problem description

Because sometimes tensor The amount of data is large ,print Output tensor The data in the middle part will be replaced by ellipsis （ Here's the picture ）, So that you can't see all the parameter sizes .

Solution

Add the following content at the beginning of the file

import torch
torch.set_printoptions(threshold=np.inf)