th> a=torch.zeros(1,5)
                                                                      [0.0001s]
th> a
 0  0  0  0  0
[torch.DoubleTensor of size 1x5]

                                                                      [0.0001s]
th> a[{1,floor}]=1
                                                                      [0.0001s]
th> a
 1  1  1  1  1
[torch.DoubleTensor of size 1x5]

                                                                      [0.0001s]
th> a[{1,ceil}]=0.5
                                                                      [0.0001s]
th> a
 0.5000  0.5000  0.5000  0.5000  0.5000
[torch.DoubleTensor of size 1x5]

 

相关文章: