使用new_zeros和zeros_like分别返回与输入相同尺寸/类型/device的tensor和ndarray

news/2024/10/25 11:32:48/
  • torch.Tensor.new_zeros(size, dtype=None, device=None)

返回尺寸为size的全为0的tensor,默认,返回的tensor与该tensor具有相同的dtype和device,可以用于在模型训练过程中创建新tensor,并保证该tensor在对应的device上
1)size: list或者tuple,定义输出tensor的尺寸
2)dtype:返回tensor的期望类型,默认与输入tensor具有相同的torch.dtype
3)device:返回tensor的期望device,默认与输入tensor具有相同的torch.device

  • 例如:
tensor = torch.tensor((), dtype=torch.float64)
tensor.new_zeros((2, 3))
# tensor([[ 0.,  0.,  0.],
#         [ 0.,  0.,  0.]], dtype=torch.float64)
  • numpy.zeros_like(a, dtype=None, order=‘K’, subok=True, shape=None)

返回与给定数组具有相同形状和类型的零数组。
a:输入的数组
dtype:输出数组的期望类型,默认与a相同
shape:输出数组的期望尺寸,默认与a相同

  • 例如:
x = np.arange(6)
x = x.reshape((2, 3))
# x
# array([[0, 1, 2],
#        [3, 4, 5]])
np.zeros_like(x)
# array([[0, 0, 0],
#        [0, 0, 0]])

http://www.ppmy.cn/news/1200026.html

相关文章

Go 中的反射

本教程有以下部分。 什么是反射?需要检查变量并查找其类型吗?反射包 反射类型和反射值反射.KindNumField() 和 Field() 方法Int() 和 String() 方法 完整的程序是否应该使用反射? 现在让我们一一讨论这些部分。 什么是反射? 反…

办公套件全家桶 Office2019 mac中文版新功能

office 2019 mac是 Microsoft office 应用程序套件的最新版本。它包括流行的软件,例如 Microsoft Word、Excel、PowerPoint 和 Outlook,office 2019 比其前身有许多新功能和改进,包括增强的协作工具、与 OneDrive 和 SharePoint 等云服务的更…

由一个自动化脚本运维展开的思考

今天分享一个思路,如何通过脚本集中管理程序的启停。减少人工的介入。 例子 好的,这里有一个基本的shell脚本示例,你可以根据你的具体需求进行修改。 启动脚本(start.sh): #!/bin/bash ./test_server_1…

ACTIVE_MQ学习

ActiveMq学习①___入门概述https://blog.csdn.net/qq_45905724/article/details/131796502 ActiveMq学习②__安装与控制台https://blog.csdn.net/qq_45905724/article/details/133893214 ActiveMq学习③___Java编码实现ActiveMQ通讯https://blog.csdn.net/qq_45905724/articl…

每个程序员都应该自己写一个的:socket包装类

每个程序员都应该有自己的网络类。 下面是我自己用的socket类,支持所有我自己常用的功能,支持windows和unix/linux。 目录 客户端 服务端 非阻塞 获取socket信息 完整代码 客户端 作为socket客户端,只需要如下几个功能: //…

SM5102 3.7V 锂电池转干电池充放管理芯片

SM5102 3.7V 锂电池转干电池充放管理芯片 简介 : SM5102 是一款锂电池充放电管理专用芯片。充电工作时, 可以为 3.7V 锂电池进行充电,电流最高可配置1A。放电工作时,采用开关频率 1MHz同步降压转换器进行放电,放电电流可以达到3…

CATIA环境编辑器用不了时创建项目快捷方式

CATIA环境编辑器用不了时创建项目快捷方式 一、参考适用情况示例二、 解决步骤(一) 先正确放置winb_64部署包(二) 添加环境文件(三) 修改加入的环境文件(四) 复制本机CATIA快捷方式后重命名(五) 修改快捷方式目标的值 一、参考适用情况示例 二、 解决步骤 (一) 先正确放置winb…

分享与SOHO朋友聊天的一些经验

昨天跟一个同是SOHO的朋友见面聊了一下,她说自己的困境是不知道如何提升工作效率,感觉一个人自律性太差了,甚至有时候客户来找都提不起精神来跟进,需要抱团,希望能一起相互鼓励一下,问我有没有什么方法&…