Searched refs:axis_position (Results 1 – 2 of 2) sorted by relevance
218 def pack(labeled_tensors, new_axis, axis_position=0, name=None): argument254 [t.tensor for t in labeled_tensors], axis=axis_position, name=scope)256 axes.insert(axis_position, new_axis)
198 [self.original_lt, self.original_lt], new_axis='batch', axis_position=4)