torch.tensor(data, dtype=None, device=None):根据指定数据创建张量,可自动推断数据类型,支持指定数据类型和设备。
torch.Tensor(data):根据形状创建浮点张量,或根据输入数据创建张量,会强制将数据转为torch.float32。
torch.linspace(start, end, steps):在start和end之间均匀生成steps个元素的线性张量。
torch.randn(*sizes):生成服从标准正态分布的随机张量。
torch.manual_seed(seed):设置随机数种子,确保随机操作的可重复性。
torch.zeros_like(input, dtype=None, device=None):创建全0张量,前者根据指定形状创建,后者根据输入张量的形状创建。
torch.full_like(input, fill_value, dtype=None, device=None):创建全为指定值的张量。
torch.eye(n, m=None, dtype=None, device=None):创建主对角线上为1的单位张量
20万+

被折叠的 条评论
为什么被折叠?



