def squeeze(x, axis=None): if axis is None: # define axis axis = [] x_shape = int_shape(x) for s in range(len(x_shape)): if x_shape[s] == 1: axis.append(s) if type(axis) is list: x_squeezed = x for i in range(len(axis)): ax = axis[i] - i x_squeezed = _KerasNode('squeeze', tensor=plaidml_op.squeeze(x_squeezed.tensor, ax)) return x_squeezed else: return _KerasNode('squeeze', tensor=plaidml_op.squeeze(x.tensor, axis))
def squeeze(x, axis=None): if axis is None: # Auto-squeeze the size 1 dims. Note that this never squeezes symbolic dims axis = [] x_shape = int_shape(x) for s in range(len(x_shape)): if x_shape[s] == 1: axis.append(s) return _KerasNode('squeeze', tensor=plaidml_op.squeeze(x.tensor, axis))
def squeeze(x, axis): logger.debug('squeeze(x: {}, axis: {})'.format(x, axis)) return _KerasNode('squeeze', tensor=plaidml_op.squeeze(x.tensor, axis))
def squeeze(x, axis): return _KerasNode('squeeze', tensor=plaidml_op.squeeze(x.tensor, axis))