Skip to content

Instantly share code, notes, and snippets.

@rian-dolphin
Created November 3, 2021 21:42
Show Gist options
  • Save rian-dolphin/ba954d932919f6e71f1af0cffb123597 to your computer and use it in GitHub Desktop.
Save rian-dolphin/ba954d932919f6e71f1af0cffb123597 to your computer and use it in GitHub Desktop.
train_scaled = torch.FloatTensor(train_scaled)
print(f'Original dimensions : {train_scaled.shape}')
train_scaled = train_scaled.view(-1)
print(f'Correct dimensions : {train_scaled.shape}')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment