Python – PyTorch floor() 方法介绍

简介

floor() 是 PyTorch 中的数学函数之一，用于对张量中的元素进行逐个取下限操作，返回一个新的张量。它可以应用于整数、浮点数或复数类型。本篇文章将向大家介绍 floor() 方法的用法、参数以及示例。

语法格式

torch.floor(input, out=None) -> Tensor

参数说明

• input：被操作的输入张量。

• out（可选）：输出张量。

返回值

该方法执行完后，返回一个新的张量，其包含与 input 张量相同形状的张量，其中每个元素被替换为其下限的值。

示例

示例 1

import torch

x = torch.tensor([3.3, -0.8, 2.7, -1.5, 4.9])
y = torch.floor(x)

print(y)

输出结果

tensor([ 3., -1.,  2., -2.,  4.])

示例 2

import torch

x = torch.tensor([1, 2, 3, 4, 5])
y = torch.floor(x)

print(y)

输出结果

tensor([1, 2, 3, 4, 5])

示例 3

import torch

x = torch.tensor([1+2j, 3-4j])
y = torch.floor(x)

print(y)

输出结果

RuntimeError: floor is not implemented for complex tensors

在第三个示例中，将抛出一个错误，因为 floor() 方法不支持复杂张量的操作。