charlieoneill11/split.py

Created January 11, 2022 00:24

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/a5f27cf87be574e8535de93c17cdf665.js"></script>
Save charlieoneill11/a5f27cf87be574e8535de93c17cdf665 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

	a = torch.from_numpy(y[3:, :-1])
	b = a.split(1, dim=1)
	b[0].shape

	>>> torch.Size([97, 1])

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