inspect.py

train_data = train.train_data
train_data = train.transform(train_data.numpy())
# print(train_data[:, 0, :].shape)
# for px in train_data[:, 0, :]:
#   print(px, ' ')

print('[Train]')
print(' - Numpy Shape:', train.train_data.cpu().numpy().shape)
print(' - Tensor Shape:', train.train_data.size())
print(' - min:', torch.min(train_data))
print(' - max:', torch.max(train_data))
print(' - mean:', torch.mean(train_data))
print(' - std:', torch.std(train_data))
print(' - var:', torch.var(train_data))