public void DiagMatAxis_InvalidAxis() { // arrange var axis1 = 1; var axis2 = 3; var input = NdArray <int> .Arange(HostDevice.Instance, 0, 10, 1).Reshape(new[] { 2, 5 }); // action var diagMat = NdArrayOperator <int> .DiagMatAxis(axis1, axis2, input); }
public void DiagMatAxis() { // arrange var axis1 = 0; var axis2 = 1; var input = NdArray <int> .Arange(HostDevice.Instance, 0, 12, 1).Reshape(new[] { 4, 3 }); // action var diagMat = NdArrayOperator <int> .DiagMatAxis(axis1, axis2, input); // assert CollectionAssert.AreEqual(new[] { 4, 4, 3 }, diagMat.Shape); }