张量填充。

1.张量填充

函数torch.full()返回一个大小为fill_value的张量的张量。

import torch

a = torch.full([3, 4], 3)
print("a = ", a)

b = torch.full([2, 5], 3.5)
print("b = ", b)

运行结果：

a =  tensor([[3, 3, 3, 3],
        [3, 3, 3, 3],
        [3, 3, 3, 3]])
b =  tensor([[3.5000, 3.5000, 3.5000, 3.5000, 3.5000],
        [3.5000, 3.5000, 3.5000, 3.5000, 3.5000]])