forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathNDArray.Equal.cs
More file actions
41 lines (38 loc) · 1.28 KB
/
NDArray.Equal.cs
File metadata and controls
41 lines (38 loc) · 1.28 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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());
}
}
}