charlieoneill11/train_shape.py

Created January 11, 2022 00:25

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/charlieoneill11/5be8ecbb24f7516a31eeb16f5f04117b.js"></script>
Save charlieoneill11/5be8ecbb24f7516a31eeb16f5f04117b to your computer and use it in GitHub Desktop.

Download ZIP

Raw

train_shape.py

	# y = (100, 1000)
	train_input = torch.from_numpy(y[3:, :-1]) # (97, 999)
	train_target = torch.from_numpy(y[3:, 1:]) # (97, 999)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment