Merge pull request #69 from sugarme/ts

change package 'tensor' to 'ts'

dutil/dataloader.go

@@ -4,7 +4,7 @@ import (
 	"fmt"
 	"reflect"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // DataLoader combines a dataset and a sampler and provides

dutil/dataset_test.go

@@ -85,7 +85,7 @@ func TestNewMapDataset(t *testing.T) {
 	}
 }
 
-func TestMaptDataset_Len(t *testing.T) {
+func TestMapDataset_Len(t *testing.T) {
 	var data map[string]int = map[string]int{"one": 1, "two": 2}
 	ds, err := dutil.NewMapDataset(data)
 	if err != nil {

example/augmentation/main.go

@@ -7,7 +7,6 @@ import (
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/vision"
 	"github.com/sugarme/gotch/vision/aug"
-	// ts "github.com/sugarme/gotch/tensor"
 )
 
 func main() {

example/basic/main.go

@@ -4,7 +4,7 @@ import (
 	"fmt"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func main() {

example/char-rnn/main.go

@@ -6,7 +6,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 const (

example/cifar/main.go

@@ -14,7 +14,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/convert-model/main.go

@@ -4,7 +4,7 @@ import (
 	"fmt"
 	"log"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func main() {

example/debug-memory/main.go

@@ -6,7 +6,7 @@ import (
 	"log"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 var device string

example/jit-train/main.go

@@ -7,7 +7,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/jit/main.go

@@ -10,7 +10,7 @@ import (
 	"log"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/mnist/cnn.go

@@ -7,7 +7,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/mnist/linear.go

@@ -4,7 +4,7 @@ import (
 	"fmt"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/mnist/nn.go

@@ -6,7 +6,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/neural-style-transfer/main.go

@@ -12,7 +12,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/pretrained-model/main.go

@@ -12,7 +12,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/scheduler/main.go

@@ -6,8 +6,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
-	// "github.com/sugarme/gotch/vision"
+	"github.com/sugarme/gotch/ts"
 )
 
 func main() {

example/tensor-grad/main.go

@@ -2,18 +2,17 @@ package main
 
 import (
 	"fmt"
-	// "log"
 
-	"github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func main() {
-	x := tensor.TensorFrom([]float64{2.0})
+	x := ts.TensorFrom([]float64{2.0})
 	x = x.MustSetRequiresGrad(true, false)
 	x.ZeroGrad()
 
-	xy := tensor.TensorFrom([]float64{2.0})
-	xz := tensor.TensorFrom([]float64{3.0})
+	xy := ts.TensorFrom([]float64{2.0})
+	xz := ts.TensorFrom([]float64{3.0})
 
 	y := x.MustMul(xy, false)
 	z := x.MustMul(xz, false)
@@ -25,9 +24,9 @@ func main() {
 	xgrad = x.MustGrad(false)
 	xgrad.Print() // [5.0] due to accumulated 2.0 + 3.0
 
-	isGradEnabled := tensor.MustGradSetEnabled(false)
+	isGradEnabled := ts.MustGradSetEnabled(false)
 	fmt.Printf("Previous GradMode enabled state: %v\n", isGradEnabled)
-	isGradEnabled = tensor.MustGradSetEnabled(true)
+	isGradEnabled = ts.MustGradSetEnabled(true)
 	fmt.Printf("Previous GradMode enabled state: %v\n", isGradEnabled)
 
 }

example/tensor-io/main.go

@@ -4,45 +4,45 @@ import (
 	"fmt"
 
 	"github.com/sugarme/gotch"
-	"github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func main() {
 
-	ts, err := tensor.OfSlice([]float64{1.3, 29.7})
+	x, err := ts.OfSlice([]float64{1.3, 29.7})
 	if err != nil {
 		panic(err)
 	}
 
 	path := "file.pt"
-	ts.MustSave(path)
+	x.MustSave(path)
 
-	loadedTs := tensor.MustLoad(path)
+	loadedTs := ts.MustLoad(path)
 
 	loadedTs.Print()
 
-	ts1 := tensor.MustOfSlice([]float64{1.3, 29.7})
-	ts2 := tensor.MustOfSlice([]float64{2.1, 31.2})
+	ts1 := ts.MustOfSlice([]float64{1.3, 29.7})
+	ts2 := ts.MustOfSlice([]float64{2.1, 31.2})
 
-	var namedTensors []tensor.NamedTensor = []tensor.NamedTensor{
+	var namedTensors []ts.NamedTensor = []ts.NamedTensor{
 		{Name: "ts1", Tensor: ts1},
 		{Name: "ts2", Tensor: ts2},
 	}
 
 	pathMulti := "file_multi.pt"
 
-	// err = tensor.SaveMulti(namedTensors, pathMulti)
+	// err = ts.SaveMulti(namedTensors, pathMulti)
 	// if err != nil {
 	// panic(err)
 	// }
-	err = tensor.SaveMultiNew(namedTensors, pathMulti)
+	err = ts.SaveMultiNew(namedTensors, pathMulti)
 	if err != nil {
 		panic(err)
 	}
 
-	var data []tensor.NamedTensor
+	var data []ts.NamedTensor
 
-	data = tensor.MustLoadMulti(pathMulti)
+	data = ts.MustLoadMulti(pathMulti)
 
 	for _, v := range data {
 		v.Tensor.Print()
@@ -50,18 +50,18 @@ func main() {
 
 	device := gotch.NewCuda()
 
-	data = tensor.MustLoadMultiWithDevice(pathMulti, device)
+	data = ts.MustLoadMultiWithDevice(pathMulti, device)
 	for _, v := range data {
 		v.Tensor.Print()
 	}
 
-	tsString := ts.MustToString(80)
+	tsString := x.MustToString(80)
 
 	fmt.Printf("Tensor String: \n%v\n", tsString)
 
 	imagePath := "mnist-sample.png"
 
-	imageTs, err := tensor.LoadHwc(imagePath)
+	imageTs, err := ts.LoadHwc(imagePath)
 	if err != nil {
 		panic(err)
 	}

example/transfer-learning/main.go

@@ -11,7 +11,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

example/translation/main.go

@@ -19,7 +19,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 var (

example/yolo/darknet.go

@@ -11,7 +11,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type Block struct {

example/yolo/draw.go

@@ -3,6 +3,7 @@ package main
 import (
 	"image"
 	"image/color"
+
 	// "image/jpeg"
 	"io/ioutil"
 
@@ -15,7 +16,7 @@ import (
 	"golang.org/x/image/font"
 
 	"github.com/sugarme/gotch/example/yolo/freetype"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 var (

example/yolo/main.go

@@ -10,7 +10,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

gen/gen.ml

@@ -758,7 +758,7 @@ let write_cpp funcs filename =
 let write_wrapper funcs filename =
   Out_channel.with_file filename ~f:(fun out_ml ->
       let pm s = print_inline out_ml s in
-      pm "package tensor" ;
+      pm "package ts" ;
       pm "\n\n" ;
       pm "// NOTE. THIS FILE IS AUTOMATICALLY GENERATED, DO NOT EDIT BY HAND!" ;
       pm "\n\n" ;
@@ -989,7 +989,7 @@ let write_wrapper funcs filename =
 let write_must_wrapper funcs filename =
   Out_channel.with_file filename ~f:(fun out_ml ->
       let pm s = print_inline out_ml s in
-      pm "package tensor" ;
+      pm "package ts" ;
       pm "\n\n" ;
       pm "// NOTE. THIS FILE IS AUTOMATICALLY GENERATED, DO NOT EDIT BY HAND!" ;
       pm "\n\n" ;
@@ -1347,5 +1347,5 @@ let () =
   run ~yaml_filename:"gen/pytorch/Declarations-v1.10.0.yaml"
     ~cpp_filename:"libtch/torch_api_generated"
     ~ffi_filename:"libtch/c-generated.go"
-    ~must_wrapper_filename:"tensor/must-tensor-generated.go"
-    ~wrapper_filename:"tensor/tensor-generated.go"
+    ~must_wrapper_filename:"ts/must-tensor-generated.go"
+    ~wrapper_filename:"ts/tensor-generated.go"

nn/batch-norm.go

@@ -5,7 +5,7 @@ package nn
 import (
 	"log"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Batch-normalization config.

nn/conv-transpose.go

@@ -5,7 +5,7 @@ package nn
 import (
 	"log"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type ConvTranspose1DConfig struct {

nn/conv.go

@@ -6,7 +6,7 @@ import (
 	"fmt"
 	"reflect"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Conv1DConfig:

nn/func.go

@@ -3,7 +3,7 @@ package nn
 // Layers defined by closure
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type Func struct {

nn/init.go

@@ -5,7 +5,7 @@ import (
 	"math"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type Init interface {

nn/jit.go

@@ -5,7 +5,7 @@ import (
 	"log"
 	"strings"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // TrainableCModule is a trainable version of JIT Pytorch module

nn/layer-norm.go

@@ -2,7 +2,7 @@ package nn
 
 // A layer-normalization layer.
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Layer-normalization config.

nn/linear.go

@@ -6,7 +6,7 @@ import (
 	"math"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // LinearConfig is a configuration for a linear layer

nn/loss.go

@@ -2,7 +2,7 @@ package nn
 
 import (
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type lossFnOptions struct {

nn/optimizer.go

@@ -8,7 +8,7 @@ import (
 	"math"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Optimizer is a struct object to run gradient descent.

nn/optimizer_test.go

@@ -6,7 +6,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func TestOptimizer(t *testing.T) {

nn/other.go

@@ -1,7 +1,7 @@
 package nn
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Dropout:

nn/rnn.go

@@ -4,7 +4,7 @@ import (
 	"fmt"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type State interface{}

nn/rnn_test.go

@@ -7,7 +7,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func gruTest(rnnConfig *nn.RNNConfig, t *testing.T) {

nn/sequential.go

@@ -4,8 +4,7 @@ package nn
 
 import (
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
-	// "reflect"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Sequential is a layer (container) that combines multiple other layers.

nn/sparse.go

@@ -3,7 +3,7 @@ package nn
 // Sparse layers
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Configuration option for an embedding layer.

nn/sparse_test.go

@@ -6,7 +6,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func embeddingTest(embeddingConfig *nn.EmbeddingConfig, t *testing.T) {

nn/varstore.go

@@ -9,7 +9,7 @@ import (
 	"sync"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // SEP is a separator to separate path elements in the tensor names.

nn/varstore_test.go

@@ -8,7 +8,7 @@ import (
 
 	"github.com/sugarme/gotch"
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func TestVarStoreEntry(t *testing.T) {

pickle/serialization.go

@@ -21,7 +21,7 @@ import (
 	"sort"
 
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 const hexMagicNumber = "1950a86a20f9469cfc6c"

ts/basic-example_test.go

@@ -1,10 +1,10 @@
-package tensor_test
+package ts_test
 
 import (
 	"fmt"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func ExampleTensor_MustArange() {

ts/data.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 import (
 	"fmt"

ts/data_test.go

@@ -1,4 +1,4 @@
-package tensor_test
+package ts_test
 
 import (
 	// "fmt"
@@ -9,7 +9,7 @@ import (
 	"testing"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func TestTextData_NewTextData(t *testing.T) {

ts/error.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 // #include <stdlib.h>
 import "C"

ts/image.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 import "C"
 

ts/index.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 // Indexing operations for tensor
 // It defines a `i` indexing operation. This can be used in various scenarios.

ts/index_test.go

@@ -1,11 +1,11 @@
-package tensor_test
+package ts_test
 
 import (
 	"reflect"
 	"testing"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func TestIntegerIndex(t *testing.T) {

ts/iter.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 import (
 	"fmt"

ts/jit.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 // JIT interface to run model trained/saved using PyTorch Python API.
 

ts/jit_test.go

@@ -1,10 +1,10 @@
-package tensor_test
+package ts_test
 
 import (
 	"reflect"
 	"testing"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func roundTrip(v interface{}, t *testing.T) {

ts/module.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 // Module interface is a container with only one method `Forward`
 //

ts/npy.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 import (
 	"archive/zip"

ts/npy_test.go

@@ -1,11 +1,11 @@
-package tensor_test
+package ts_test
 
 import (
 	"reflect"
 	"testing"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func TestNpyHeaderParse(t *testing.T) {

ts/optimizer.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 import (
 	"log"

ts/other.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 // Other tensor methods
 

ts/patch-example_test.go

@@ -1,10 +1,10 @@
-package tensor_test
+package ts_test
 
 import (
 	"testing"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func ExampleTensor_Split(t *testing.T) {

ts/patch.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 // #include "stdlib.h"
 import "C"

ts/print.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 import (
 	"bytes"

ts/scalar.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 import (
 	// "unsafe"

ts/tensor.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 //#include "stdlib.h"
 //#include "stdbool.h"

ts/tensor_test.go

@@ -1,11 +1,11 @@
-package tensor_test
+package ts_test
 
 import (
 	"reflect"
 	"testing"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func TestTensorInit(t *testing.T) {

ts/util.go

@@ -1,4 +1,4 @@
-package tensor
+package ts
 
 // #include <stdlib.h>
 import "C"

vision/alexnet.go

@@ -2,7 +2,7 @@ package vision
 
 import (
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // AlexNet implementation

vision/aug/affine.go

@@ -2,7 +2,7 @@ package aug
 
 import (
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // RandomAffine is transformation of the image keeping center invariant.

vision/aug/blur.go

@@ -5,7 +5,7 @@ import (
 	"log"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type GaussianBlur struct {

vision/aug/color.go

@@ -4,7 +4,7 @@ import (
 	"fmt"
 	"log"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Ref. https://github.com/pytorch/vision/blob/f1d734213af65dc06e777877d315973ba8386080/torchvision/transforms/functional_tensor.py

vision/aug/contrast.go

@@ -1,7 +1,7 @@
 package aug
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // RandomAutocontrast autocontrasts the pixels of the given image randomly with a given probability.

vision/aug/crop.go

@@ -7,7 +7,7 @@ import (
 	// "math"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type RandomCrop struct {

vision/aug/cutout.go

@@ -6,7 +6,7 @@ import (
 	"math"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Randomly selects a rectangle region in an torch Tensor image and erases its pixels.

vision/aug/equalize.go

@@ -1,7 +1,7 @@
 package aug
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // RandomEqualize equalizes the histogram of the given image randomly with a given probability.

vision/aug/flip.go

@@ -2,7 +2,7 @@ package aug
 
 import (
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // RandomHorizontalFlip horizontally flips the given image randomly with a given probability.

vision/aug/function.go

@@ -8,7 +8,7 @@ import (
 	"time"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func gaussianKernel1D(ks int64, sigma float64, dtype gotch.DType, device gotch.Device) *ts.Tensor {

vision/aug/grayscale.go

@@ -3,8 +3,7 @@ package aug
 import (
 	"log"
 
-	ts "github.com/sugarme/gotch/tensor"
-	// "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // GrayScale converts image to grayscale.

vision/aug/invert.go

@@ -1,7 +1,7 @@
 package aug
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type RandomInvert struct {

vision/aug/normalize.go

@@ -1,7 +1,7 @@
 package aug
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Normalize normalizes a tensor image with mean and standard deviation.

vision/aug/perspective.go

@@ -4,7 +4,7 @@ import (
 	// "fmt"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // RandomPerspective performs a random perspective transformation of the given image with a given probability.

vision/aug/posterize.go

@@ -1,7 +1,7 @@
 package aug
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // RandomPosterize posterizes the image randomly with a given probability by reducing the

vision/aug/resize.go

@@ -5,7 +5,7 @@ import (
 	"log"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 	"github.com/sugarme/gotch/vision"
 )
 

vision/aug/rotate.go

@@ -8,7 +8,7 @@ import (
 	"time"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // RandomRotate randomly rotates a tensor image within a specifed angle range (degree).

vision/aug/sharpness.go

@@ -1,7 +1,7 @@
 package aug
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Adjust the sharpness of the image randomly with a given probability. If the image is torch Tensor,

vision/aug/solarize.go

@@ -1,7 +1,7 @@
 package aug
 
 import (
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // RandomSolarize solarizes the image randomly with a given probability by inverting all pixel

vision/aug/transform.go

@@ -5,7 +5,7 @@ import (
 	"time"
 
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Transformer is an interface that can transform an image tensor.

vision/cifar.go

@@ -13,7 +13,7 @@ import (
 	"path/filepath"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 const (

vision/dataset.go

@@ -8,7 +8,7 @@ import (
 	"math/rand"
 	"time"
 
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 type Dataset struct {

vision/densenet.go

@@ -9,7 +9,7 @@ import (
 	"fmt"
 
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func dnConv2d(p *nn.Path, cIn, cOut, ksize, padding, stride int64) *nn.Conv2D {

vision/efficientnet.go

@@ -5,7 +5,7 @@ import (
 	"math"
 
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 const (

vision/image.go

@@ -9,7 +9,7 @@ import (
 	"math"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // (height, width, channel) -> (channel, height, width)

vision/imagenet.go

@@ -11,7 +11,7 @@ import (
 	"sync"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // Helper functions for ImageNet like datasets.

vision/inception.go

@@ -4,7 +4,7 @@ package vision
 
 import (
 	"github.com/sugarme/gotch/nn"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 func convBn(p *nn.Path, cIn, cOut, ksize, pad, stride int64) ts.ModuleT {

vision/mnist.go

@@ -12,7 +12,7 @@ import (
 	"path/filepath"
 
 	"github.com/sugarme/gotch"
-	ts "github.com/sugarme/gotch/tensor"
+	"github.com/sugarme/gotch/ts"
 )
 
 // readInt32 read 4 bytes and convert to MSB first (big endian) interger.