@@ -73,6 +71,7 @@ class LinspaceOpMaker : public framework::OpProtoAndCheckerMaker {
...
@@ -73,6 +71,7 @@ class LinspaceOpMaker : public framework::OpProtoAndCheckerMaker {
AddInput("Num",
AddInput("Num",
"Number of entry in the sequence. It is a tensor of shape [1], "
"Number of entry in the sequence. It is a tensor of shape [1], "
"should be of type int32.");
"should be of type int32.");
AddAttr<int>("dtype","The output data type.");
AddOutput("Out","A sequence of numbers.");
AddOutput("Out","A sequence of numbers.");
AddComment(R"DOC(
AddComment(R"DOC(
Return fixed number of evenly spaced values within a given interval. First entry is start, and last entry is stop. In the case when Num is 1, only Start is returned. Like linspace function of numpy.
Return fixed number of evenly spaced values within a given interval. First entry is start, and last entry is stop. In the case when Num is 1, only Start is returned. Like linspace function of numpy.
...
@@ -85,4 +84,6 @@ class LinspaceOpMaker : public framework::OpProtoAndCheckerMaker {
...
@@ -85,4 +84,6 @@ class LinspaceOpMaker : public framework::OpProtoAndCheckerMaker {