element-wise multiplication

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Time    : 2020/9/24 16:50
# @Author  : zhaoyLi
# @FileName: test2.py
# @Software: PyCharm
# @Blog    :https://blog.csdn.net/qq_37405118
# @Purpose :验证element-wise multiplication


import torch

x = torch.randn(1,3,2,2)
y = torch.tensor([[[[1.0]],

         [[2.0]],

         [[3.0]]]]) #1,3,1,1
print(y.shape)
z = torch.mul(x,y)
print(x,'\n',y,'\n',z)
print(z.shape)#1,3,2,2

#############################################**输出**##########################################
E:\Anconda\python.exe "C:/Users/MR-LI/Desktop/program practice/9-24-2020/test2.py"
torch.Size([1, 3, 1, 1])
tensor([[[[ 0.6868,  0.0787],
          [ 0.4054,  0.4241]],

         [[-1.4643, -1.7135],
          [ 0.2437,  0.4013]],

         [[-0.6950,  1.0245],
          [ 0.8811,  0.7769]]]]) 
 tensor([[[[1.]],

         [[2.]],

         [[3.]]]]) 
 tensor([[[[ 0.6868,  0.0787],
          [ 0.4054,  0.4241]],

         [[-2.9287, -3.4269],
          [ 0.4874,  0.8026]],

         [[-2.0849,  3.0734],
          [ 2.6432,  2.3306]]]])
torch.Size([1, 3, 2, 2])
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值