github.com/wzzhu/tensor@v0.9.24/benchmark_dense_repeat_test.go (about)

     1  package tensor
     2  
     3  import "testing"
     4  
     5  func BenchmarkDenseRepeat(b *testing.B) {
     6  	for _, tst := range repeatTests {
     7  		tst := tst
     8  		b.Run(tst.name, func(b *testing.B) {
     9  			for n := 0; n < b.N; n++ {
    10  				tst.tensor.Repeat(tst.axis, tst.repeats...)
    11  			}
    12  		})
    13  	}
    14  }