torch.fill_()
函数是一个in-place操作,用于将张量的所有元素设置为给定的标量值。它接受一个标量参数作为输入,该标量将用于填充整个张量。
以下是一个例子:
import torch # 创建一个形状为(2, 3)的张量,并用1来填充 a = torch.empty((2, 3)).fill_(1) print(a)
上述代码中,首先使用torch.empty()
函数创建了一个形状为(2,3)的空张量a
,然后使用fill_()
函数将它的所有元素都填充为1。输出结果如下所示:
tensor([[1., 1., 1.], [1., 1., 1.]])
需要注意的是,由于torch.fill_()
函数是一个in-place操作,因此它会直接修改原始张量的值,并且不会返回任何值。如果想要保留原始张量并且生成一个新的被填充的张量,可以使用torch.full()
函数。与torch.fill_()
不同,torch.full()
函数返回一个新的张量对象,而不会修改原始张量。