Home
last modified time | relevance | path

Searched refs:new_axes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dops.py301 def reshape(labeled_tensor, existing_axes, new_axes, name=None): argument
353 if sum(isinstance(axis, string_types) for axis in new_axes) > 1:
356 'axes must have an indicated integer size or labels: %r' % new_axes)
361 for axis_ref in new_axes:
371 axes = original_values[:start] + list(new_axes) + original_values[stop:]
1022 new_axes = [
1026 return core.LabeledTensor(tile_op, new_axes)
1062 new_axes = []
1069 new_axes.append(core.concat_axes([axis_before, axis, axis_after]))
1072 new_axes.append(axis)
[all …]
Dsugar.py56 def __init__(self, existing_axis_names, new_axes, name=None): argument
59 self._new_axes = new_axes
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py277 new_axes = {4: [0, 3, 1, 2], 5: [0, 4, 1, 2, 3]}
280 return array_ops.transpose(input_tensor, new_axes[ndims])
283 return [input_tensor[a] for a in new_axes[ndims]]
363 new_axes = {4: [0, 2, 3, 1], 5: [0, 2, 3, 4, 1]}
366 return array_ops.transpose(input_tensor, new_axes[ndims])
369 return [input_tensor[a] for a in new_axes[ndims]]