Skip to content

Commit 9534cbb

Browse files
committed
Added zeros, reshape, add, add_, addbmm, baddbmm, exp, mul, mul_, pow, sigmoid, sub initUniform
1 parent 7759c63 commit 9534cbb

File tree

4 files changed

+1017
-131
lines changed

4 files changed

+1017
-131
lines changed

Test/TorchSharp.cs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -399,16 +399,15 @@ public void TestSubInPlace()
399399
var x = IntTensor.Ones(new long[] { 100, 100 });
400400
var y = IntTensor.Ones(new long[] { 100, 100 });
401401

402-
var z = x.SubInPlace(y);
402+
x.SubInPlace(y);
403403

404-
var zdata = z.Data;
405404
var xdata = x.Data;
406405

407406
for (int i = 0; i < 100; i++)
408407
{
409408
for (int j = 0; j < 100; j++)
410409
{
411-
Assert.AreEqual(zdata[i + j], xdata[i + j]);
410+
Assert.AreEqual(0, xdata[i + j]);
412411
}
413412
}
414413
}

TorchSharp/Tensor/ITorchTensor.cs

Lines changed: 27 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,16 +30,40 @@ public interface ITorchTensor<T> : IDisposable
3030

3131
ITorchTensor<float> Grad();
3232

33+
ITorchTensor<T> Reshape(params long[] shape);
34+
3335
ITorchTensor<T> View(params long[] shape);
3436

3537
ITorchTensor<U> Eq<U>(ITorchTensor<U> target);
3638

37-
ITorchTensor<T> SubInPlace(ITorchTensor<T> target);
39+
ITorchTensor<T> Add(ITorchTensor<T> target, int scalar);
3840

39-
ITorchTensor<T> Mul(T scalar);
41+
void AddInPlace(ITorchTensor<T> target, int scalar);
4042

41-
ITorchTensor<T> Sum();
43+
ITorchTensor<T> Addbmm(ITorchTensor<T> batch1, ITorchTensor<T> batch2, float beta, float alpha);
4244

4345
ITorchTensor<T> Argmax(long dimension, bool keepDimension = false);
46+
47+
ITorchTensor<T> Baddbmm(ITorchTensor<T> batch2, ITorchTensor<T> mat, float beta, float alpha);
48+
49+
ITorchTensor<T> Exp();
50+
51+
ITorchTensor<T> MatMul(ITorchTensor<T> target);
52+
53+
ITorchTensor<T> Mul(ITorchTensor<T> target);
54+
55+
ITorchTensor<T> Mul(T scalar);
56+
57+
void MulInPlace(ITorchTensor<T> target);
58+
59+
ITorchTensor<T> Pow(float scalar);
60+
61+
ITorchTensor<T> Sigmoid();
62+
63+
ITorchTensor<T> Sub(ITorchTensor<T> target);
64+
65+
void SubInPlace(ITorchTensor<T> target);
66+
67+
ITorchTensor<T> Sum();
4468
}
4569
}

0 commit comments

Comments
 (0)