Crate caffe2op_shape

source ·

Structs

  • ShapeOp
    | Produce a 1D int64 tensor with the shape | of the input tensor. | | If called with an optional argument | axes, the result will only contain | the dimensions of specified axes. | | Github Link: | | - https://github.com/pytorch/pytorch/blob/master/caffe2/operators/shape_op.cc | | RecordShapeOp records the shape of | the input tensor to a vector of int. You | mostly don’t need this operator explicitly, | and it is mostly used in the autodiff | process. |