Skip to content

Commit 96f8370

Browse files
authored
Merge pull request dotnet#27 from interesaaat/LibTorchSharpFirstTest
Add eq and add with scalars
2 parents d7fc585 + a1c07bc commit 96f8370

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

TorchSharp/Tensor/TorchTensor.cs

+21
Original file line numberDiff line numberDiff line change
@@ -403,6 +403,14 @@ public TorchTensor Add(TorchTensor target, int scalar = 1)
403403
return new TorchTensor(THSTensor_add(handle, scalar, target.Handle));
404404
}
405405

406+
[DllImport("libTorchSharp")]
407+
extern static IntPtr THSTensor_addS(IntPtr src, IntPtr trg);
408+
409+
public TorchTensor Add(Scalar scalar)
410+
{
411+
return new TorchTensor(THSTensor_addS(handle, scalar.Handle));
412+
}
413+
406414
[DllImport("libTorchSharp")]
407415
extern static void THSTensor_add_(IntPtr src, int scalar, IntPtr trg);
408416

@@ -483,6 +491,14 @@ public TorchTensor Eq(TorchTensor target)
483491
return new TorchTensor(THSTensor_eq(handle, target.Handle));
484492
}
485493

494+
[DllImport("libTorchSharp")]
495+
extern static IntPtr THSTensor_eqS(IntPtr src, IntPtr trg);
496+
497+
public TorchTensor Eq(Scalar target)
498+
{
499+
return new TorchTensor(THSTensor_eq(handle, target.Handle));
500+
}
501+
486502
[DllImport("libTorchSharp")]
487503
extern static bool THSTensor_equal(IntPtr src, IntPtr trg);
488504

@@ -616,6 +632,11 @@ public TorchTensor Sum(long[] dimensions, bool keepDimension = false)
616632
return left.Add(right);
617633
}
618634

635+
public static TorchTensor operator +(TorchTensor left, Scalar right)
636+
{
637+
return left.Add(right);
638+
}
639+
619640
public static TorchTensor operator *(TorchTensor left, TorchTensor right)
620641
{
621642
return left.Mul(right);

0 commit comments

Comments
 (0)