How to use 'concatenate tensors pytorch' in Python

Every line of 'concatenate tensors pytorch' code snippets is scanned for vulnerabilities by our powerful machine learning engine that combs millions of open source libraries, ensuring your Python code is secure.

All examples are scanned by Snyk Code

By copying the Snyk Code Snippets you agree to

ezyang/onnx-pytorch

104 def _flatten_tensors(tensors):
105     """Flatten tensors into a single contiguous 1D buffer"""
106     if len(tensors) == 1:
107         return tensors[0].contiguous().view(-1)
108     numels = [tensor.numel() for tensor in tensors]
109     size = sum(numels)
110     offset = 0
111     flat = tensors[0].new(size)
112     for tensor, numel in zip(tensors, numels):
113         flat.narrow(0, offset, numel).copy_(tensor, broadcast=False)
114         offset += numel
115     return flat

104	def _flatten_tensors(tensors):
105	"""Flatten tensors into a single contiguous 1D buffer"""
106	if len(tensors) == 1:
107	return tensors[0].contiguous().view(-1)
108	numels = [tensor.numel() for tensor in tensors]
109	size = sum(numels)
110	offset = 0
111	flat = tensors[0].new(size)
112	for tensor, numel in zip(tensors, numels):
113	flat.narrow(0, offset, numel).copy_(tensor, broadcast=False)
114	offset += numel
115	return flat

How to use 'concatenate tensors pytorch' in Python

All examples are scanned by Snyk Code

Related snippets