TensorFlow.NET/src/TensorFlowNET.Core/NumPy/NDArray.Equal.cs at master · ehtick/TensorFlow.NET · GitHub

41 lines (38 loc) · 1.28 KB

using System;

using System.Collections.Generic;

using System.Linq;

using System.Text;

using static Tensorflow.Binding;

namespace Tensorflow.NumPy

{

public partial class NDArray

{

public override bool Equals(object obj)

{

return obj switch

{

int val => GetAtIndex<int>(0) == val,

long val => GetAtIndex<long>(0) == val,

float val => GetAtIndex<float>(0) == val,

double val => GetAtIndex<double>(0) == val,

string val => StringData(0) == val,

int[] val => ToArray<int>().SequenceEqual(val),

long[] val => ToArray<long>().SequenceEqual(val),

float[] val => ToArray<float>().SequenceEqual(val),

double[] val => ToArray<double>().SequenceEqual(val),

NDArray val => Equals(this, val),

_ => base.Equals(obj)

};

}

bool Equals(NDArray x, NDArray y)

{

if (x.ndim != y.ndim)

return false;

else if (x.size != y.size)

return false;

else if (x.dtype != y.dtype)

return false;

return Enumerable.SequenceEqual(x.ToByteArray(), y.ToByteArray());

}

Expand file tree