pytorch torch.full_like函数介绍

torch.full_like 是 PyTorch 中用于创建一个具有特定值的新张量,其形状和数据类型与给定张量相同。

函数定义

复制代码
torch.full_like(input, fill_value, *, dtype=None, layout=None, device=None, requires_grad=False, memory_format=torch.preserve_format)

参数说明

  1. input (Tensor):

    • 用来提供形状和其他属性(如设备、数据类型等)的参考张量。
  2. fill_value (float 或 int):

    • 用于填充新张量的值。
  3. dtype (torch.dtype, 可选):

    • 新张量的数据类型。如果未指定,则与 input 的数据类型相同。
  4. layout (torch.layout, 可选):

    • 新张量的内存布局。默认为 input 的布局。
  5. device (torch.device, 可选):

    • 新张量所在的设备。如果未指定,则与 input 的设备相同。
  6. requires_grad (bool, 可选, 默认值:False):

    • 如果为 True,新张量将需要梯度计算。
  7. memory_format (torch.memory_format, 可选):

    • 新张量的内存格式。默认为 torch.preserve_format,即与 input 相同的内存格式。

返回值

  • 返回一个新张量,其形状、设备、数据类型等与 input 相同,但所有元素均为 fill_value

示例

1. 基本用法
复制代码
import torch

# 创建一个参考张量
x = torch.tensor([[1, 2], [3, 4]])

# 创建一个与 x 形状相同的新张量,元素全为 5
result = torch.full_like(x, 5)
print(result)
# tensor([[5, 5],
#         [5, 5]])
2. 指定数据类型
复制代码
result = torch.full_like(x, 5.0, dtype=torch.float32)
print(result)
# tensor([[5.0, 5.0],
#         [5.0, 5.0]])
3. 指定设备
复制代码
result = torch.full_like(x, 3, device='cuda')
print(result)  # 张量在 GPU 上
4. 需要梯度
复制代码
result = torch.full_like(x, 2, requires_grad=True)
print(result.requires_grad)  # True

常见用途

  1. 快速初始化张量:在网络初始化、测试时创建具有固定值的张量。
  2. 占位符:生成形状与参考张量相同的占位张量。
  3. 兼容性计算:确保新张量与给定张量具有相同的数据类型和设备。

注意事项

  • torch.full 的区别torch.full 需要手动指定张量的形状,而 torch.full_like 自动使用参考张量的形状。

  • 支持广播fill_value 可以是标量。

  • 性能优化torch.full_like 会自动优化设备和数据类型,便于高效地创建张量。

相关推荐
非门由也4 小时前
《sklearn机器学习——管道和复合估计器》联合特征(FeatureUnion)
人工智能·机器学习·sklearn
l12345sy4 小时前
Day21_【机器学习—决策树(1)—信息增益、信息增益率、基尼系数】
人工智能·决策树·机器学习·信息增益·信息增益率·基尼指数
非门由也4 小时前
《sklearn机器学习——管道和复合估算器》异构数据的列转换器
人工智能·机器学习·sklearn
计算机毕业设计指导4 小时前
基于ResNet50的智能垃圾分类系统
人工智能·分类·数据挖掘
飞哥数智坊4 小时前
终端里用 Claude Code 太难受?我把它接进 TRAE,真香!
人工智能·claude·trae
java1234_小锋5 小时前
Scikit-learn Python机器学习 - 特征降维 压缩数据 - 特征提取 - 主成分分析 (PCA)
python·机器学习·scikit-learn
java1234_小锋5 小时前
Scikit-learn Python机器学习 - 特征降维 压缩数据 - 特征提取 - 线性判别分析 (LDA)
python·机器学习·scikit-learn
小王爱学人工智能5 小时前
OpenCV的阈值处理
人工智能·opencv·计算机视觉
新智元5 小时前
刚刚,光刻机巨头 ASML 杀入 AI!豪掷 15 亿押注「欧版 OpenAI」,成最大股东
人工智能·openai
机器之心5 小时前
全球图生视频榜单第一,爱诗科技PixVerse V5如何改变一亿用户的视频创作
人工智能·openai