Function mindspore::NN::Input

Function Documentation

std::unique_ptr<Node> mindspore::NN::Input(std::vector<int> dims, DataType data_type = DataType::kNumberTypeFloat32, int fmt = NHWC)