NumCpp  2.5.1
A Templatized Header Only C++ Implementation of the Python NumPy Library
identity.hpp
Go to the documentation of this file.
1 #pragma once
29 
30 #include "NumCpp/NdArray.hpp"
31 
32 namespace nc
33 {
34  //============================================================================
35  // Method Description:
48  template<typename dtype>
50  {
51  NdArray<dtype> returnArray(inSquareSize);
52  returnArray.zeros();
53  for (uint32 i = 0; i < inSquareSize; ++i)
54  {
55  returnArray(i, i) = dtype{ 1 };
56  }
57 
58  return returnArray;
59  }
60 } // namespace nc
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:72
NdArray< dtype > & zeros() noexcept
Definition: NdArrayCore.hpp:4848
Definition: Coordinate.hpp:45
NdArray< dtype > identity(uint32 inSquareSize)
Definition: identity.hpp:49
std::uint32_t uint32
Definition: Types.hpp:40