Exemplo n.º 1
0
        public void AvgPool2DTensor()
        {
            TorchTensor ones = Float32Tensor.Ones(new long[] { 4, 2, 2, 2 });
            var         obj  = ones.AvgPool2D(new long[] { 2, 2 });

            Assert.Equal(typeof(TorchTensor), obj.GetType());
            Assert.Equal(Float32Tensor.Ones(new long[] { 4, 2, 1, 1 }), obj);
        }