torch.minimum
时间: 2023-10-20 17:32:35 浏览: 112
Minim
torch.minimum是一个torch库中的函数,用于计算两个输入张量中的元素最小值。它将两个张量作为输入,并返回一个新的张量,其中每个元素是对应位置上两个输入张量中元素的较小值。[1]
例如,如果我们有两个输入张量a和b:
a = torch.tensor((1, 2, -1))
b = torch.tensor((3, 0, 4))
那么torch.minimum(a, b)将返回一个新的张量,其中每个元素是a和b对应位置上的较小值:
tensor([1, 0, -1])
在这个示例中,a和b的第一个元素分别是1和3,所以在新的张量中,第一个元素为1,以此类推。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [Pytorch速成教程(二) 常用函数](https://blog.csdn.net/HUASHUDEYANJING/article/details/126920065)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文