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))