Exemplos de unsqueeze em Python

Exemplo n.º 1

Arquivo: shape_inference_test.py Projeto: hansely/NNEF-Tools

 def test_unsqueeze(self):
     self.assertEqual([1, 2, 3], infer.unsqueeze([2, 3], [0]))
     self.assertEqual([2, 3, 1], infer.unsqueeze([2, 3], [-1]))
     self.assertEqual([2, 1, 3], infer.unsqueeze([2, 3], [-2]))
     self.assertEqual([1, 2, 3], infer.unsqueeze([2, 3], [-3]))
     self.assertEqual([1, 2, 1], infer.unsqueeze([1, 2], [2]))
     self.assertEqual([2, 1, 3], infer.unsqueeze([2, 3], [1]))
     self.assertEqual([1, 1, 1, 2, 1, 1], infer.unsqueeze([1, 2], [0, 2, 4, 5]))
     self.assertEqual([1], infer.unsqueeze([], [0]))

Exemplo n.º 2

Arquivo: operations.py Projeto: stjordanis/NNEF-Tools

def nnef_unsqueeze(input, axes):
    return input.reshape(shape_inference.unsqueeze(input.shape, axes))

Exemplo n.º 3

Arquivo: tf_pb_shape_inference.py Projeto: benmbark/NNEF-Tools

def propagate_expand_dims(op, const_value_by_tensor):
    # type: (TFOperation, _ConstValueByTensorT)->typing.Tuple[typing.List[typing.List[int]], typing.List[str]]
    input, axis = op.inputs
    axis = const_value_by_tensor[axis].item()
    return [infer.unsqueeze(input=input.shape, axes=[axis])], [op.attribs['T']]

Exemplo n.º 4

def propagate_unsqueeze(op):
    # type: (ONNXOperation)->typing.Tuple[typing.List[typing.List[int]], typing.List[str]]

    return [infer.unsqueeze(input=op.input.shape,
                            axes=op.attribs['axes'])], [op.input.dtype]

Exemplo n.º 5

Arquivo: caffe2_shapes.py Projeto: stjordanis/NNEF-Tools

def expand_dims_shape(op):
    # type: (Caffe2Operation)->ShapeResult
    return infer.unsqueeze(op.inputs[0].shape,
                           axes=op.attribs['dims']), op.inputs[0].dtype