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);
}
