Skip to content

Wrong broadcastable output of Alloc when using SpecifyShape #692

Closed
@ricardoV94

Description

@ricardoV94
import aesara.tensor as at

constant_size = at.constant([1])
specify_size = at.specify_shape(constant_size, [1])

print(constant_size.type)  # TensorType(int32, (True,))
print(specify_size.type)  # TensorType(int32, (True,))

print(at.alloc(0, *constant_size).type)  # TensorType(int8, (True,))
print(at.alloc(0, *specify_size).type)  # TensorType(int8, vector)

Metadata

Metadata

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions